├── .gitignore ├── README.md ├── environment.sh ├── figures ├── case_study.png ├── logo.png └── pipeline.png ├── requirements.txt ├── run_dpo_train.sh ├── run_testing.sh ├── run_train_testing.sh ├── run_training.sh ├── test ├── PDEcontrol │ ├── .gitignore │ ├── cog.yaml │ └── evaluation │ │ ├── data_processing │ │ ├── __init__.py │ │ ├── answer_extraction.py │ │ └── process_utils.py │ │ ├── eval │ │ ├── __init__.py │ │ ├── eval_robustness_reasoning_wrapper.py │ │ ├── eval_robustness_wrapper.py │ │ ├── eval_script.py │ │ └── utils.py │ │ ├── infer │ │ └── simulate_gt.py │ │ └── scripts │ │ └── infer_pdecontrol.sh ├── README.md ├── requirements.txt └── scripts │ ├── read_result.py │ ├── simulate_gt.sh │ └── test_pdecontrol.sh ├── train ├── README.md ├── config │ ├── deepspeed.json │ └── deepspeed_dpo.json ├── scripts │ ├── group_text.py │ ├── group_text_dpo.py │ ├── merge_model.py │ ├── tokenize_data.py │ ├── tokenize_data_dpo.py │ ├── train.sh │ ├── train_dpo.sh │ └── utils │ │ ├── loader.py │ │ ├── trainer.py │ │ └── util.py ├── train.py ├── train_dpo.py ├── train_finetune.py ├── utils │ ├── loader.py │ ├── trainer.py │ └── util.py └── validate.py ├── train_test_combined.yml └── utils └── few_shot_prompts ├── __init__.py ├── cot_one_d_combined_fewshot.py ├── cot_one_d_heat_fewshot.py ├── cot_one_d_wave_fewshot.py ├── examples ├── DPO_one_d_combined │ ├── dataset_description.json │ └── examples.jsonl ├── DPO_one_d_heat │ ├── dataset_description.json │ └── examples.jsonl ├── DPO_one_d_wave │ ├── dataset_description.json │ └── examples.jsonl ├── one_d_combined │ ├── dataset_description.json │ └── examples.jsonl ├── one_d_heat │ ├── dataset_description.json │ └── examples.jsonl └── one_d_wave │ ├── dataset_description.json │ └── examples.jsonl ├── few_shot_prompting.py ├── few_shot_test.py ├── few_shot_train.py └── few_shot_train_dpo.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/README.md -------------------------------------------------------------------------------- /environment.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/environment.sh -------------------------------------------------------------------------------- /figures/case_study.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/figures/case_study.png -------------------------------------------------------------------------------- /figures/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/figures/logo.png -------------------------------------------------------------------------------- /figures/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/figures/pipeline.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_dpo_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/run_dpo_train.sh -------------------------------------------------------------------------------- /run_testing.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/run_testing.sh -------------------------------------------------------------------------------- /run_train_testing.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/run_train_testing.sh -------------------------------------------------------------------------------- /run_training.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/run_training.sh -------------------------------------------------------------------------------- /test/PDEcontrol/.gitignore: -------------------------------------------------------------------------------- 1 | *__pycache__ 2 | -------------------------------------------------------------------------------- /test/PDEcontrol/cog.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/PDEcontrol/cog.yaml -------------------------------------------------------------------------------- /test/PDEcontrol/evaluation/data_processing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /test/PDEcontrol/evaluation/data_processing/answer_extraction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/PDEcontrol/evaluation/data_processing/answer_extraction.py -------------------------------------------------------------------------------- /test/PDEcontrol/evaluation/data_processing/process_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/PDEcontrol/evaluation/data_processing/process_utils.py -------------------------------------------------------------------------------- /test/PDEcontrol/evaluation/eval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/PDEcontrol/evaluation/eval/__init__.py -------------------------------------------------------------------------------- /test/PDEcontrol/evaluation/eval/eval_robustness_reasoning_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/PDEcontrol/evaluation/eval/eval_robustness_reasoning_wrapper.py -------------------------------------------------------------------------------- /test/PDEcontrol/evaluation/eval/eval_robustness_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/PDEcontrol/evaluation/eval/eval_robustness_wrapper.py -------------------------------------------------------------------------------- /test/PDEcontrol/evaluation/eval/eval_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/PDEcontrol/evaluation/eval/eval_script.py -------------------------------------------------------------------------------- /test/PDEcontrol/evaluation/eval/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/PDEcontrol/evaluation/eval/utils.py -------------------------------------------------------------------------------- /test/PDEcontrol/evaluation/infer/simulate_gt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/PDEcontrol/evaluation/infer/simulate_gt.py -------------------------------------------------------------------------------- /test/PDEcontrol/evaluation/scripts/infer_pdecontrol.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/PDEcontrol/evaluation/scripts/infer_pdecontrol.sh -------------------------------------------------------------------------------- /test/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/README.md -------------------------------------------------------------------------------- /test/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/requirements.txt -------------------------------------------------------------------------------- /test/scripts/read_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/scripts/read_result.py -------------------------------------------------------------------------------- /test/scripts/simulate_gt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/scripts/simulate_gt.sh -------------------------------------------------------------------------------- /test/scripts/test_pdecontrol.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/test/scripts/test_pdecontrol.sh -------------------------------------------------------------------------------- /train/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/README.md -------------------------------------------------------------------------------- /train/config/deepspeed.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/config/deepspeed.json -------------------------------------------------------------------------------- /train/config/deepspeed_dpo.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/config/deepspeed_dpo.json -------------------------------------------------------------------------------- /train/scripts/group_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/scripts/group_text.py -------------------------------------------------------------------------------- /train/scripts/group_text_dpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/scripts/group_text_dpo.py -------------------------------------------------------------------------------- /train/scripts/merge_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/scripts/merge_model.py -------------------------------------------------------------------------------- /train/scripts/tokenize_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/scripts/tokenize_data.py -------------------------------------------------------------------------------- /train/scripts/tokenize_data_dpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/scripts/tokenize_data_dpo.py -------------------------------------------------------------------------------- /train/scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/scripts/train.sh -------------------------------------------------------------------------------- /train/scripts/train_dpo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/scripts/train_dpo.sh -------------------------------------------------------------------------------- /train/scripts/utils/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/scripts/utils/loader.py -------------------------------------------------------------------------------- /train/scripts/utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/scripts/utils/trainer.py -------------------------------------------------------------------------------- /train/scripts/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/scripts/utils/util.py -------------------------------------------------------------------------------- /train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/train.py -------------------------------------------------------------------------------- /train/train_dpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/train_dpo.py -------------------------------------------------------------------------------- /train/train_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/train_finetune.py -------------------------------------------------------------------------------- /train/utils/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/utils/loader.py -------------------------------------------------------------------------------- /train/utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/utils/trainer.py -------------------------------------------------------------------------------- /train/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/utils/util.py -------------------------------------------------------------------------------- /train/validate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train/validate.py -------------------------------------------------------------------------------- /train_test_combined.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/train_test_combined.yml -------------------------------------------------------------------------------- /utils/few_shot_prompts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/__init__.py -------------------------------------------------------------------------------- /utils/few_shot_prompts/cot_one_d_combined_fewshot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/cot_one_d_combined_fewshot.py -------------------------------------------------------------------------------- /utils/few_shot_prompts/cot_one_d_heat_fewshot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/cot_one_d_heat_fewshot.py -------------------------------------------------------------------------------- /utils/few_shot_prompts/cot_one_d_wave_fewshot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/cot_one_d_wave_fewshot.py -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/DPO_one_d_combined/dataset_description.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/DPO_one_d_combined/dataset_description.json -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/DPO_one_d_combined/examples.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/DPO_one_d_combined/examples.jsonl -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/DPO_one_d_heat/dataset_description.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/DPO_one_d_heat/dataset_description.json -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/DPO_one_d_heat/examples.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/DPO_one_d_heat/examples.jsonl -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/DPO_one_d_wave/dataset_description.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/DPO_one_d_wave/dataset_description.json -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/DPO_one_d_wave/examples.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/DPO_one_d_wave/examples.jsonl -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/one_d_combined/dataset_description.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/one_d_combined/dataset_description.json -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/one_d_combined/examples.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/one_d_combined/examples.jsonl -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/one_d_heat/dataset_description.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/one_d_heat/dataset_description.json -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/one_d_heat/examples.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/one_d_heat/examples.jsonl -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/one_d_wave/dataset_description.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/one_d_wave/dataset_description.json -------------------------------------------------------------------------------- /utils/few_shot_prompts/examples/one_d_wave/examples.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/examples/one_d_wave/examples.jsonl -------------------------------------------------------------------------------- /utils/few_shot_prompts/few_shot_prompting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/few_shot_prompting.py -------------------------------------------------------------------------------- /utils/few_shot_prompts/few_shot_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/few_shot_test.py -------------------------------------------------------------------------------- /utils/few_shot_prompts/few_shot_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/few_shot_train.py -------------------------------------------------------------------------------- /utils/few_shot_prompts/few_shot_train_dpo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/delta-lab-ai/pde-controller/HEAD/utils/few_shot_prompts/few_shot_train_dpo.py --------------------------------------------------------------------------------