├── .gitignore ├── LICENSE ├── README.md ├── applications ├── coal_mining │ ├── __init__.py │ ├── configuration.py │ ├── custom_network.py │ ├── data.py │ ├── experiments.py │ ├── likelihoods.py │ ├── models.py │ ├── notebooks │ │ └── bayesloop_benchmark.ipynb │ ├── plots │ │ └── coal_mining_benchmark.png │ ├── posterior_samples │ │ ├── .DS_Store │ │ ├── bl_post_means.npy │ │ └── bl_post_stds.npy │ └── priors.py ├── lexical_decision │ ├── data │ │ ├── data_lexical_decision.csv │ │ └── parameters_full_ddm_error_coding_cs.lst │ ├── notebooks │ │ ├── full_gpddm.ipynb │ │ ├── full_rwddm.ipynb │ │ ├── mmd.ipynb │ │ └── sbc.ipynb │ └── src │ │ ├── context.py │ │ ├── helpers.py │ │ ├── macro_models.py │ │ ├── micro_models.py │ │ ├── networks.py │ │ ├── plot_funs.py │ │ ├── priors.py │ │ └── transformations.py ├── optimal_policy │ ├── configuration.py │ ├── data │ │ ├── data_preprocessing.R │ │ └── optimal_policy_data.csv │ ├── experiments.py │ ├── notebook │ │ └── optimal_policy_experiment.ipynb │ └── plots │ │ └── optimal_policy_result.png ├── simulation_study │ ├── configuration.py │ ├── experiments.py │ └── notebook │ │ └── simulation_study_experiment.ipynb ├── stan_comparison │ ├── .DS_Store │ ├── __init__.py │ ├── checkpoints │ │ └── stan_benchmark │ │ │ ├── checkpoint │ │ │ ├── ckpt-50.data-00000-of-00001 │ │ │ ├── ckpt-50.index │ │ │ └── history_50.pkl │ ├── configuration.py │ ├── experiments.py │ ├── legacy │ │ ├── models.py │ │ └── priors.py │ ├── likelihoods.py │ ├── models.py │ ├── notebooks │ │ └── stan_benchmark.ipynb │ ├── plots │ │ └── stan_benchmark.png │ └── priors.py └── version.py ├── assets └── diffusion │ ├── __init__.py │ ├── likelihoods.py │ ├── models.py │ └── priors.py ├── environment.yaml ├── param_recovery_animation.gif ├── pyproject.toml ├── setup.cfg └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/README.md -------------------------------------------------------------------------------- /applications/coal_mining/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /applications/coal_mining/configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/configuration.py -------------------------------------------------------------------------------- /applications/coal_mining/custom_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/custom_network.py -------------------------------------------------------------------------------- /applications/coal_mining/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/data.py -------------------------------------------------------------------------------- /applications/coal_mining/experiments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/experiments.py -------------------------------------------------------------------------------- /applications/coal_mining/likelihoods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/likelihoods.py -------------------------------------------------------------------------------- /applications/coal_mining/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/models.py -------------------------------------------------------------------------------- /applications/coal_mining/notebooks/bayesloop_benchmark.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/notebooks/bayesloop_benchmark.ipynb -------------------------------------------------------------------------------- /applications/coal_mining/plots/coal_mining_benchmark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/plots/coal_mining_benchmark.png -------------------------------------------------------------------------------- /applications/coal_mining/posterior_samples/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/posterior_samples/.DS_Store -------------------------------------------------------------------------------- /applications/coal_mining/posterior_samples/bl_post_means.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/posterior_samples/bl_post_means.npy -------------------------------------------------------------------------------- /applications/coal_mining/posterior_samples/bl_post_stds.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/posterior_samples/bl_post_stds.npy -------------------------------------------------------------------------------- /applications/coal_mining/priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/coal_mining/priors.py -------------------------------------------------------------------------------- /applications/lexical_decision/data/data_lexical_decision.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/data/data_lexical_decision.csv -------------------------------------------------------------------------------- /applications/lexical_decision/data/parameters_full_ddm_error_coding_cs.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/data/parameters_full_ddm_error_coding_cs.lst -------------------------------------------------------------------------------- /applications/lexical_decision/notebooks/full_gpddm.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/notebooks/full_gpddm.ipynb -------------------------------------------------------------------------------- /applications/lexical_decision/notebooks/full_rwddm.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/notebooks/full_rwddm.ipynb -------------------------------------------------------------------------------- /applications/lexical_decision/notebooks/mmd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/notebooks/mmd.ipynb -------------------------------------------------------------------------------- /applications/lexical_decision/notebooks/sbc.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/notebooks/sbc.ipynb -------------------------------------------------------------------------------- /applications/lexical_decision/src/context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/src/context.py -------------------------------------------------------------------------------- /applications/lexical_decision/src/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/src/helpers.py -------------------------------------------------------------------------------- /applications/lexical_decision/src/macro_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/src/macro_models.py -------------------------------------------------------------------------------- /applications/lexical_decision/src/micro_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/src/micro_models.py -------------------------------------------------------------------------------- /applications/lexical_decision/src/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/src/networks.py -------------------------------------------------------------------------------- /applications/lexical_decision/src/plot_funs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/src/plot_funs.py -------------------------------------------------------------------------------- /applications/lexical_decision/src/priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/src/priors.py -------------------------------------------------------------------------------- /applications/lexical_decision/src/transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/lexical_decision/src/transformations.py -------------------------------------------------------------------------------- /applications/optimal_policy/configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/optimal_policy/configuration.py -------------------------------------------------------------------------------- /applications/optimal_policy/data/data_preprocessing.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/optimal_policy/data/data_preprocessing.R -------------------------------------------------------------------------------- /applications/optimal_policy/data/optimal_policy_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/optimal_policy/data/optimal_policy_data.csv -------------------------------------------------------------------------------- /applications/optimal_policy/experiments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/optimal_policy/experiments.py -------------------------------------------------------------------------------- /applications/optimal_policy/notebook/optimal_policy_experiment.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/optimal_policy/notebook/optimal_policy_experiment.ipynb -------------------------------------------------------------------------------- /applications/optimal_policy/plots/optimal_policy_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/optimal_policy/plots/optimal_policy_result.png -------------------------------------------------------------------------------- /applications/simulation_study/configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/simulation_study/configuration.py -------------------------------------------------------------------------------- /applications/simulation_study/experiments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/simulation_study/experiments.py -------------------------------------------------------------------------------- /applications/simulation_study/notebook/simulation_study_experiment.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/simulation_study/notebook/simulation_study_experiment.ipynb -------------------------------------------------------------------------------- /applications/stan_comparison/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/.DS_Store -------------------------------------------------------------------------------- /applications/stan_comparison/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /applications/stan_comparison/checkpoints/stan_benchmark/checkpoint: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/checkpoints/stan_benchmark/checkpoint -------------------------------------------------------------------------------- /applications/stan_comparison/checkpoints/stan_benchmark/ckpt-50.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/checkpoints/stan_benchmark/ckpt-50.data-00000-of-00001 -------------------------------------------------------------------------------- /applications/stan_comparison/checkpoints/stan_benchmark/ckpt-50.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/checkpoints/stan_benchmark/ckpt-50.index -------------------------------------------------------------------------------- /applications/stan_comparison/checkpoints/stan_benchmark/history_50.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/checkpoints/stan_benchmark/history_50.pkl -------------------------------------------------------------------------------- /applications/stan_comparison/configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/configuration.py -------------------------------------------------------------------------------- /applications/stan_comparison/experiments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/experiments.py -------------------------------------------------------------------------------- /applications/stan_comparison/legacy/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/legacy/models.py -------------------------------------------------------------------------------- /applications/stan_comparison/legacy/priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/legacy/priors.py -------------------------------------------------------------------------------- /applications/stan_comparison/likelihoods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/likelihoods.py -------------------------------------------------------------------------------- /applications/stan_comparison/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/models.py -------------------------------------------------------------------------------- /applications/stan_comparison/notebooks/stan_benchmark.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/notebooks/stan_benchmark.ipynb -------------------------------------------------------------------------------- /applications/stan_comparison/plots/stan_benchmark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/plots/stan_benchmark.png -------------------------------------------------------------------------------- /applications/stan_comparison/priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/applications/stan_comparison/priors.py -------------------------------------------------------------------------------- /applications/version.py: -------------------------------------------------------------------------------- 1 | __version__ = "1.0" -------------------------------------------------------------------------------- /assets/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /assets/diffusion/likelihoods.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/assets/diffusion/likelihoods.py -------------------------------------------------------------------------------- /assets/diffusion/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/assets/diffusion/models.py -------------------------------------------------------------------------------- /assets/diffusion/priors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/assets/diffusion/priors.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/environment.yaml -------------------------------------------------------------------------------- /param_recovery_animation.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/param_recovery_animation.gif -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bayesflow-org/Neural-Superstatistics/HEAD/setup.py --------------------------------------------------------------------------------