├── .gitattributes ├── .github └── workflows │ ├── formatting.yml │ └── run_unittests.yml ├── .gitignore ├── LICENSE ├── README.md ├── gatsbi ├── networks │ ├── __init__.py │ ├── base.py │ ├── models.py │ └── modules.py ├── optimize │ ├── __init__.py │ ├── base.py │ ├── sequential.py │ ├── unrolled.py │ └── utils.py ├── task_utils │ ├── benchmarks │ │ ├── __init__.py │ │ ├── make_networks.py │ │ └── make_results.py │ ├── camera_model │ │ ├── __init__.py │ │ ├── networks.py │ │ ├── prior.py │ │ └── simulator.py │ ├── run_utils.py │ └── shallow_water_model │ │ ├── __init__.py │ │ ├── dataloader.py │ │ ├── networks.py │ │ ├── prior.py │ │ ├── sbc_analysis.py │ │ ├── shallow_water01_modified.f90 │ │ └── simulator.py └── utils │ ├── __init__.py │ ├── classifier.py │ ├── load_data.py │ └── loss_funcns.py ├── plotting_code ├── benchmark_problems.ipynb ├── camera_model.ipynb ├── matplotlibrc ├── plots │ ├── Figure2.pdf │ ├── Figure3.pdf │ ├── Figure4.pdf │ ├── Figure5.pdf │ ├── SuppFigure6.pdf │ ├── SuppFigure7.png │ ├── SuppFigure8.png │ └── SuppFigure9.pdf ├── plotting_data │ ├── camera_samples.npz │ ├── gatsbi_slcp_samples.npy │ ├── gatsbi_two_moons_samples.npy │ ├── shallow_water_data.npz │ ├── slcp_c2st_scores.yaml │ └── two_moons_c2st_scores.yaml └── shallow_water.ipynb ├── quickstart.ipynb ├── run_benchmarks.py ├── run_highdim_applications.py ├── run_inference_nle.py ├── run_inference_npe.py ├── run_inference_nre.py ├── sample_shallow_water.py ├── setup.py ├── tasks ├── camera_model │ └── defaults.yaml ├── shallow_water_model │ ├── defaults.yaml │ └── shallow_water01.pdf ├── slcp │ ├── defaults.yaml │ └── sweep_config.yaml ├── two_moons │ ├── defaults.yaml │ ├── defaults_opt.yaml │ ├── sweep_config.yaml │ └── sweep_config_LFVI.yaml └── two_moons_seq │ ├── defaults.yaml │ └── sweep_config.yaml ├── test_requirements.txt └── tests ├── pytest.ini ├── test_amortized_optimizer.py ├── test_classifier.py ├── test_dataloader.py ├── test_generator_and_discriminator_networks.py └── test_sequential_optimizer_w_corrfactor.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/workflows/formatting.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/.github/workflows/formatting.yml -------------------------------------------------------------------------------- /.github/workflows/run_unittests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/.github/workflows/run_unittests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/README.md -------------------------------------------------------------------------------- /gatsbi/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/networks/__init__.py -------------------------------------------------------------------------------- /gatsbi/networks/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/networks/base.py -------------------------------------------------------------------------------- /gatsbi/networks/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/networks/models.py -------------------------------------------------------------------------------- /gatsbi/networks/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/networks/modules.py -------------------------------------------------------------------------------- /gatsbi/optimize/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/optimize/__init__.py -------------------------------------------------------------------------------- /gatsbi/optimize/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/optimize/base.py -------------------------------------------------------------------------------- /gatsbi/optimize/sequential.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/optimize/sequential.py -------------------------------------------------------------------------------- /gatsbi/optimize/unrolled.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/optimize/unrolled.py -------------------------------------------------------------------------------- /gatsbi/optimize/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/optimize/utils.py -------------------------------------------------------------------------------- /gatsbi/task_utils/benchmarks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/benchmarks/__init__.py -------------------------------------------------------------------------------- /gatsbi/task_utils/benchmarks/make_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/benchmarks/make_networks.py -------------------------------------------------------------------------------- /gatsbi/task_utils/benchmarks/make_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/benchmarks/make_results.py -------------------------------------------------------------------------------- /gatsbi/task_utils/camera_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/camera_model/__init__.py -------------------------------------------------------------------------------- /gatsbi/task_utils/camera_model/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/camera_model/networks.py -------------------------------------------------------------------------------- /gatsbi/task_utils/camera_model/prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/camera_model/prior.py -------------------------------------------------------------------------------- /gatsbi/task_utils/camera_model/simulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/camera_model/simulator.py -------------------------------------------------------------------------------- /gatsbi/task_utils/run_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/run_utils.py -------------------------------------------------------------------------------- /gatsbi/task_utils/shallow_water_model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/shallow_water_model/__init__.py -------------------------------------------------------------------------------- /gatsbi/task_utils/shallow_water_model/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/shallow_water_model/dataloader.py -------------------------------------------------------------------------------- /gatsbi/task_utils/shallow_water_model/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/shallow_water_model/networks.py -------------------------------------------------------------------------------- /gatsbi/task_utils/shallow_water_model/prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/shallow_water_model/prior.py -------------------------------------------------------------------------------- /gatsbi/task_utils/shallow_water_model/sbc_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/shallow_water_model/sbc_analysis.py -------------------------------------------------------------------------------- /gatsbi/task_utils/shallow_water_model/shallow_water01_modified.f90: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/shallow_water_model/shallow_water01_modified.f90 -------------------------------------------------------------------------------- /gatsbi/task_utils/shallow_water_model/simulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/task_utils/shallow_water_model/simulator.py -------------------------------------------------------------------------------- /gatsbi/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/utils/__init__.py -------------------------------------------------------------------------------- /gatsbi/utils/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/utils/classifier.py -------------------------------------------------------------------------------- /gatsbi/utils/load_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/utils/load_data.py -------------------------------------------------------------------------------- /gatsbi/utils/loss_funcns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/gatsbi/utils/loss_funcns.py -------------------------------------------------------------------------------- /plotting_code/benchmark_problems.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/benchmark_problems.ipynb -------------------------------------------------------------------------------- /plotting_code/camera_model.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/camera_model.ipynb -------------------------------------------------------------------------------- /plotting_code/matplotlibrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/matplotlibrc -------------------------------------------------------------------------------- /plotting_code/plots/Figure2.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plots/Figure2.pdf -------------------------------------------------------------------------------- /plotting_code/plots/Figure3.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plots/Figure3.pdf -------------------------------------------------------------------------------- /plotting_code/plots/Figure4.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plots/Figure4.pdf -------------------------------------------------------------------------------- /plotting_code/plots/Figure5.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plots/Figure5.pdf -------------------------------------------------------------------------------- /plotting_code/plots/SuppFigure6.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plots/SuppFigure6.pdf -------------------------------------------------------------------------------- /plotting_code/plots/SuppFigure7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plots/SuppFigure7.png -------------------------------------------------------------------------------- /plotting_code/plots/SuppFigure8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plots/SuppFigure8.png -------------------------------------------------------------------------------- /plotting_code/plots/SuppFigure9.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plots/SuppFigure9.pdf -------------------------------------------------------------------------------- /plotting_code/plotting_data/camera_samples.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plotting_data/camera_samples.npz -------------------------------------------------------------------------------- /plotting_code/plotting_data/gatsbi_slcp_samples.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plotting_data/gatsbi_slcp_samples.npy -------------------------------------------------------------------------------- /plotting_code/plotting_data/gatsbi_two_moons_samples.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plotting_data/gatsbi_two_moons_samples.npy -------------------------------------------------------------------------------- /plotting_code/plotting_data/shallow_water_data.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plotting_data/shallow_water_data.npz -------------------------------------------------------------------------------- /plotting_code/plotting_data/slcp_c2st_scores.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plotting_data/slcp_c2st_scores.yaml -------------------------------------------------------------------------------- /plotting_code/plotting_data/two_moons_c2st_scores.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/plotting_data/two_moons_c2st_scores.yaml -------------------------------------------------------------------------------- /plotting_code/shallow_water.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/plotting_code/shallow_water.ipynb -------------------------------------------------------------------------------- /quickstart.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/quickstart.ipynb -------------------------------------------------------------------------------- /run_benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/run_benchmarks.py -------------------------------------------------------------------------------- /run_highdim_applications.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/run_highdim_applications.py -------------------------------------------------------------------------------- /run_inference_nle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/run_inference_nle.py -------------------------------------------------------------------------------- /run_inference_npe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/run_inference_npe.py -------------------------------------------------------------------------------- /run_inference_nre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/run_inference_nre.py -------------------------------------------------------------------------------- /sample_shallow_water.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/sample_shallow_water.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/setup.py -------------------------------------------------------------------------------- /tasks/camera_model/defaults.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tasks/camera_model/defaults.yaml -------------------------------------------------------------------------------- /tasks/shallow_water_model/defaults.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tasks/shallow_water_model/defaults.yaml -------------------------------------------------------------------------------- /tasks/shallow_water_model/shallow_water01.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tasks/shallow_water_model/shallow_water01.pdf -------------------------------------------------------------------------------- /tasks/slcp/defaults.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tasks/slcp/defaults.yaml -------------------------------------------------------------------------------- /tasks/slcp/sweep_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tasks/slcp/sweep_config.yaml -------------------------------------------------------------------------------- /tasks/two_moons/defaults.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tasks/two_moons/defaults.yaml -------------------------------------------------------------------------------- /tasks/two_moons/defaults_opt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tasks/two_moons/defaults_opt.yaml -------------------------------------------------------------------------------- /tasks/two_moons/sweep_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tasks/two_moons/sweep_config.yaml -------------------------------------------------------------------------------- /tasks/two_moons/sweep_config_LFVI.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tasks/two_moons/sweep_config_LFVI.yaml -------------------------------------------------------------------------------- /tasks/two_moons_seq/defaults.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tasks/two_moons_seq/defaults.yaml -------------------------------------------------------------------------------- /tasks/two_moons_seq/sweep_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tasks/two_moons_seq/sweep_config.yaml -------------------------------------------------------------------------------- /test_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/test_requirements.txt -------------------------------------------------------------------------------- /tests/pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tests/pytest.ini -------------------------------------------------------------------------------- /tests/test_amortized_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tests/test_amortized_optimizer.py -------------------------------------------------------------------------------- /tests/test_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tests/test_classifier.py -------------------------------------------------------------------------------- /tests/test_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tests/test_dataloader.py -------------------------------------------------------------------------------- /tests/test_generator_and_discriminator_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tests/test_generator_and_discriminator_networks.py -------------------------------------------------------------------------------- /tests/test_sequential_optimizer_w_corrfactor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mackelab/gatsbi/HEAD/tests/test_sequential_optimizer_w_corrfactor.py --------------------------------------------------------------------------------