├── README.md └── code ├── cifar10 ├── calc.py ├── cifar_input.py ├── eval.sh ├── mcbn.sh ├── resnet_main.py ├── resnet_model.py └── run.sh ├── mcbn ├── .dockerignore ├── .gitignore ├── 01. Dataset splitter.ipynb ├── 02. lambda, batch size, dropout grid search.ipynb ├── 03. Get best grid search results.ipynb ├── 04. Tau optimization.ipynb ├── 05. Get best tau results.ipynb ├── 06. Test set evaluation.ipynb ├── 07. Collect-test-results.ipynb ├── Dockerfile ├── Makefile ├── data │ ├── bostonHousing │ │ ├── housing.data │ │ ├── index_features.txt │ │ └── index_target.txt │ ├── concrete │ │ ├── concrete.data │ │ ├── index_features.txt │ │ └── index_target.txt │ ├── energy │ │ ├── ENB2012.data │ │ ├── index_features.txt │ │ └── index_target.txt │ ├── kin8nm │ │ ├── Dataset.data │ │ ├── index_features.txt │ │ └── index_target.txt │ ├── power-plant │ │ ├── index_features.txt │ │ ├── index_target.txt │ │ └── pp.data │ ├── protein-tertiary-structure │ │ ├── CASP.data │ │ ├── index_features.txt │ │ └── index_target.txt │ ├── wine-quality-red │ │ ├── index_features.txt │ │ ├── index_target.txt │ │ └── winequality-red-1.data │ └── yacht │ │ ├── index_features.txt │ │ ├── index_target.txt │ │ └── yacht_hydrodynamics.data ├── mcbn │ ├── __init__.py │ ├── data │ │ ├── __init__.py │ │ ├── dataset.py │ │ └── dataset_loaders.py │ ├── environment │ │ ├── __init__.py │ │ └── constants.py │ ├── models │ │ ├── __init__.py │ │ ├── model.py │ │ ├── model_bn.py │ │ ├── model_components.py │ │ └── model_do.py │ └── utils │ │ ├── __init__.py │ │ ├── helper.py │ │ └── metrics.py ├── mnf │ ├── __init__.py │ ├── conv_layers.py │ ├── dense_layers.py │ ├── layers.py │ ├── norm_flows.py │ ├── utils.py │ └── wrappers.py ├── mnf_evaluation.py ├── requirements.txt ├── run_evaluation.py ├── run_mnf_evaluation.sh ├── runnb.sh └── setup.yml └── segnet ├── bayesian_segnet_camvid_deploy_mcbn_select.prototxt ├── bayesian_segnet_camvid_deploy_mcbn_select2.prototxt └── test_bayesian_segnet_mcbn_paperResults.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/README.md -------------------------------------------------------------------------------- /code/cifar10/calc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/cifar10/calc.py -------------------------------------------------------------------------------- /code/cifar10/cifar_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/cifar10/cifar_input.py -------------------------------------------------------------------------------- /code/cifar10/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/cifar10/eval.sh -------------------------------------------------------------------------------- /code/cifar10/mcbn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/cifar10/mcbn.sh -------------------------------------------------------------------------------- /code/cifar10/resnet_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/cifar10/resnet_main.py -------------------------------------------------------------------------------- /code/cifar10/resnet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/cifar10/resnet_model.py -------------------------------------------------------------------------------- /code/cifar10/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/cifar10/run.sh -------------------------------------------------------------------------------- /code/mcbn/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/.dockerignore -------------------------------------------------------------------------------- /code/mcbn/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/.gitignore -------------------------------------------------------------------------------- /code/mcbn/01. Dataset splitter.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/01. Dataset splitter.ipynb -------------------------------------------------------------------------------- /code/mcbn/02. lambda, batch size, dropout grid search.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/02. lambda, batch size, dropout grid search.ipynb -------------------------------------------------------------------------------- /code/mcbn/03. Get best grid search results.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/03. Get best grid search results.ipynb -------------------------------------------------------------------------------- /code/mcbn/04. Tau optimization.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/04. Tau optimization.ipynb -------------------------------------------------------------------------------- /code/mcbn/05. Get best tau results.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/05. Get best tau results.ipynb -------------------------------------------------------------------------------- /code/mcbn/06. Test set evaluation.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/06. Test set evaluation.ipynb -------------------------------------------------------------------------------- /code/mcbn/07. Collect-test-results.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/07. Collect-test-results.ipynb -------------------------------------------------------------------------------- /code/mcbn/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/Dockerfile -------------------------------------------------------------------------------- /code/mcbn/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/Makefile -------------------------------------------------------------------------------- /code/mcbn/data/bostonHousing/housing.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/data/bostonHousing/housing.data -------------------------------------------------------------------------------- /code/mcbn/data/bostonHousing/index_features.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/data/bostonHousing/index_features.txt -------------------------------------------------------------------------------- /code/mcbn/data/bostonHousing/index_target.txt: -------------------------------------------------------------------------------- 1 | 13 2 | -------------------------------------------------------------------------------- /code/mcbn/data/concrete/concrete.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/data/concrete/concrete.data -------------------------------------------------------------------------------- /code/mcbn/data/concrete/index_features.txt: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | 4 6 | 5 7 | 6 8 | 7 9 | -------------------------------------------------------------------------------- /code/mcbn/data/concrete/index_target.txt: -------------------------------------------------------------------------------- 1 | 8 2 | -------------------------------------------------------------------------------- /code/mcbn/data/energy/ENB2012.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/data/energy/ENB2012.data -------------------------------------------------------------------------------- /code/mcbn/data/energy/index_features.txt: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | 4 6 | 5 7 | 6 8 | 7 9 | -------------------------------------------------------------------------------- /code/mcbn/data/energy/index_target.txt: -------------------------------------------------------------------------------- 1 | 8 2 | -------------------------------------------------------------------------------- /code/mcbn/data/kin8nm/Dataset.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/data/kin8nm/Dataset.data -------------------------------------------------------------------------------- /code/mcbn/data/kin8nm/index_features.txt: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | 4 6 | 5 7 | 6 8 | 7 9 | -------------------------------------------------------------------------------- /code/mcbn/data/kin8nm/index_target.txt: -------------------------------------------------------------------------------- 1 | 8 2 | -------------------------------------------------------------------------------- /code/mcbn/data/power-plant/index_features.txt: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | -------------------------------------------------------------------------------- /code/mcbn/data/power-plant/index_target.txt: -------------------------------------------------------------------------------- 1 | 4 2 | -------------------------------------------------------------------------------- /code/mcbn/data/power-plant/pp.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/data/power-plant/pp.data -------------------------------------------------------------------------------- /code/mcbn/data/protein-tertiary-structure/CASP.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/data/protein-tertiary-structure/CASP.data -------------------------------------------------------------------------------- /code/mcbn/data/protein-tertiary-structure/index_features.txt: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | 4 6 | 5 7 | 6 8 | 7 9 | 8 10 | -------------------------------------------------------------------------------- /code/mcbn/data/protein-tertiary-structure/index_target.txt: -------------------------------------------------------------------------------- 1 | 9 2 | -------------------------------------------------------------------------------- /code/mcbn/data/wine-quality-red/index_features.txt: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | 4 6 | 5 7 | 6 8 | 7 9 | 8 10 | 9 11 | 10 12 | -------------------------------------------------------------------------------- /code/mcbn/data/wine-quality-red/index_target.txt: -------------------------------------------------------------------------------- 1 | 11 2 | -------------------------------------------------------------------------------- /code/mcbn/data/wine-quality-red/winequality-red-1.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/data/wine-quality-red/winequality-red-1.data -------------------------------------------------------------------------------- /code/mcbn/data/yacht/index_features.txt: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | 4 6 | 5 7 | -------------------------------------------------------------------------------- /code/mcbn/data/yacht/index_target.txt: -------------------------------------------------------------------------------- 1 | 6 2 | -------------------------------------------------------------------------------- /code/mcbn/data/yacht/yacht_hydrodynamics.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/data/yacht/yacht_hydrodynamics.data -------------------------------------------------------------------------------- /code/mcbn/mcbn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/mcbn/mcbn/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/mcbn/mcbn/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mcbn/data/dataset.py -------------------------------------------------------------------------------- /code/mcbn/mcbn/data/dataset_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mcbn/data/dataset_loaders.py -------------------------------------------------------------------------------- /code/mcbn/mcbn/environment/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/mcbn/mcbn/environment/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mcbn/environment/constants.py -------------------------------------------------------------------------------- /code/mcbn/mcbn/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/mcbn/mcbn/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mcbn/models/model.py -------------------------------------------------------------------------------- /code/mcbn/mcbn/models/model_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mcbn/models/model_bn.py -------------------------------------------------------------------------------- /code/mcbn/mcbn/models/model_components.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mcbn/models/model_components.py -------------------------------------------------------------------------------- /code/mcbn/mcbn/models/model_do.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mcbn/models/model_do.py -------------------------------------------------------------------------------- /code/mcbn/mcbn/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/mcbn/mcbn/utils/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mcbn/utils/helper.py -------------------------------------------------------------------------------- /code/mcbn/mcbn/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mcbn/utils/metrics.py -------------------------------------------------------------------------------- /code/mcbn/mnf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/mcbn/mnf/conv_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mnf/conv_layers.py -------------------------------------------------------------------------------- /code/mcbn/mnf/dense_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mnf/dense_layers.py -------------------------------------------------------------------------------- /code/mcbn/mnf/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mnf/layers.py -------------------------------------------------------------------------------- /code/mcbn/mnf/norm_flows.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mnf/norm_flows.py -------------------------------------------------------------------------------- /code/mcbn/mnf/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mnf/utils.py -------------------------------------------------------------------------------- /code/mcbn/mnf/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mnf/wrappers.py -------------------------------------------------------------------------------- /code/mcbn/mnf_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/mnf_evaluation.py -------------------------------------------------------------------------------- /code/mcbn/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/requirements.txt -------------------------------------------------------------------------------- /code/mcbn/run_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/run_evaluation.py -------------------------------------------------------------------------------- /code/mcbn/run_mnf_evaluation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/run_mnf_evaluation.sh -------------------------------------------------------------------------------- /code/mcbn/runnb.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/runnb.sh -------------------------------------------------------------------------------- /code/mcbn/setup.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/mcbn/setup.yml -------------------------------------------------------------------------------- /code/segnet/bayesian_segnet_camvid_deploy_mcbn_select.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/segnet/bayesian_segnet_camvid_deploy_mcbn_select.prototxt -------------------------------------------------------------------------------- /code/segnet/bayesian_segnet_camvid_deploy_mcbn_select2.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/segnet/bayesian_segnet_camvid_deploy_mcbn_select2.prototxt -------------------------------------------------------------------------------- /code/segnet/test_bayesian_segnet_mcbn_paperResults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/icml-mcbn/mcbn/HEAD/code/segnet/test_bayesian_segnet_mcbn_paperResults.py --------------------------------------------------------------------------------