├── .gitignore ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── __init__.py ├── acquisitions ├── __init__.py ├── constrained_expected_improvement.py ├── evaluate_acquisitions.py ├── expected_improvement.py ├── multitask_constrained_expected_improvement.py ├── noisy_constrained_expected_improvement.py ├── noisy_expected_improvement.py └── noisy_multitask_constrained_expected_improvement.py ├── experiments ├── __init__.py ├── config_health.py ├── config_synthetic1.py ├── config_synthetic2.py ├── data.py ├── run_optimization.py └── run_optimization_test.py ├── kernels ├── __init__.py ├── causal_coregionalize_kernel.py └── causal_kernel.py ├── methods ├── __init__.py ├── base.py ├── cbo.py ├── ccbo.py └── random.py ├── requirements.txt ├── run_experiment.ipynb ├── scm_examples ├── __init__.py ├── base.py └── scm.py ├── setup.py └── utils ├── __init__.py ├── constraints_functions.py ├── cost_utils.py ├── gp_utils.py ├── graph_functions.py ├── initialisation_utils.py ├── intervention_functions.py ├── plotting_utils.py ├── sampling_utils.py ├── sampling_utils_test.py ├── scm_utils.py ├── scm_utils_test.py ├── utilities.py └── utilities_test.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/__init__.py -------------------------------------------------------------------------------- /acquisitions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/acquisitions/__init__.py -------------------------------------------------------------------------------- /acquisitions/constrained_expected_improvement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/acquisitions/constrained_expected_improvement.py -------------------------------------------------------------------------------- /acquisitions/evaluate_acquisitions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/acquisitions/evaluate_acquisitions.py -------------------------------------------------------------------------------- /acquisitions/expected_improvement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/acquisitions/expected_improvement.py -------------------------------------------------------------------------------- /acquisitions/multitask_constrained_expected_improvement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/acquisitions/multitask_constrained_expected_improvement.py -------------------------------------------------------------------------------- /acquisitions/noisy_constrained_expected_improvement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/acquisitions/noisy_constrained_expected_improvement.py -------------------------------------------------------------------------------- /acquisitions/noisy_expected_improvement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/acquisitions/noisy_expected_improvement.py -------------------------------------------------------------------------------- /acquisitions/noisy_multitask_constrained_expected_improvement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/acquisitions/noisy_multitask_constrained_expected_improvement.py -------------------------------------------------------------------------------- /experiments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/experiments/__init__.py -------------------------------------------------------------------------------- /experiments/config_health.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/experiments/config_health.py -------------------------------------------------------------------------------- /experiments/config_synthetic1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/experiments/config_synthetic1.py -------------------------------------------------------------------------------- /experiments/config_synthetic2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/experiments/config_synthetic2.py -------------------------------------------------------------------------------- /experiments/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/experiments/data.py -------------------------------------------------------------------------------- /experiments/run_optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/experiments/run_optimization.py -------------------------------------------------------------------------------- /experiments/run_optimization_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/experiments/run_optimization_test.py -------------------------------------------------------------------------------- /kernels/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/kernels/__init__.py -------------------------------------------------------------------------------- /kernels/causal_coregionalize_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/kernels/causal_coregionalize_kernel.py -------------------------------------------------------------------------------- /kernels/causal_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/kernels/causal_kernel.py -------------------------------------------------------------------------------- /methods/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/methods/__init__.py -------------------------------------------------------------------------------- /methods/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/methods/base.py -------------------------------------------------------------------------------- /methods/cbo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/methods/cbo.py -------------------------------------------------------------------------------- /methods/ccbo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/methods/ccbo.py -------------------------------------------------------------------------------- /methods/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/methods/random.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_experiment.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/run_experiment.ipynb -------------------------------------------------------------------------------- /scm_examples/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/scm_examples/__init__.py -------------------------------------------------------------------------------- /scm_examples/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/scm_examples/base.py -------------------------------------------------------------------------------- /scm_examples/scm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/scm_examples/scm.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/setup.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/constraints_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/constraints_functions.py -------------------------------------------------------------------------------- /utils/cost_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/cost_utils.py -------------------------------------------------------------------------------- /utils/gp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/gp_utils.py -------------------------------------------------------------------------------- /utils/graph_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/graph_functions.py -------------------------------------------------------------------------------- /utils/initialisation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/initialisation_utils.py -------------------------------------------------------------------------------- /utils/intervention_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/intervention_functions.py -------------------------------------------------------------------------------- /utils/plotting_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/plotting_utils.py -------------------------------------------------------------------------------- /utils/sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/sampling_utils.py -------------------------------------------------------------------------------- /utils/sampling_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/sampling_utils_test.py -------------------------------------------------------------------------------- /utils/scm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/scm_utils.py -------------------------------------------------------------------------------- /utils/scm_utils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/scm_utils_test.py -------------------------------------------------------------------------------- /utils/utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/utilities.py -------------------------------------------------------------------------------- /utils/utilities_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/ccbo/HEAD/utils/utilities_test.py --------------------------------------------------------------------------------