├── .gitignore ├── LICENSE ├── README.md ├── diff_scm ├── __init__.py ├── configs │ ├── __init__.py │ ├── brats_configs.py │ ├── get_config.py │ └── mnist_configs.py ├── datasets │ ├── __init__.py │ ├── load_brats.py │ ├── load_mnist.py │ └── loader.py ├── models │ ├── __init__.py │ ├── gaussian_diffusion.py │ ├── losses.py │ ├── nn.py │ ├── resample.py │ ├── respace.py │ └── unet.py ├── sampling │ ├── sample_counterfactual.py │ └── sampling_utils.py ├── training │ ├── __init__.py │ ├── anticausal_classifier_train.py │ ├── main_diffusion_train.py │ └── train_util.py └── utils │ ├── __init__.py │ ├── dist_util.py │ ├── fp16_util.py │ ├── logger.py │ ├── script_util.py │ └── utils.py ├── docs ├── _config.yml ├── assets │ └── diffusion_causal_relationships.png └── index.md ├── environment.yml └── examples └── example_generate_counterfactual.ipynb /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/README.md -------------------------------------------------------------------------------- /diff_scm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diff_scm/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diff_scm/configs/brats_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/configs/brats_configs.py -------------------------------------------------------------------------------- /diff_scm/configs/get_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/configs/get_config.py -------------------------------------------------------------------------------- /diff_scm/configs/mnist_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/configs/mnist_configs.py -------------------------------------------------------------------------------- /diff_scm/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diff_scm/datasets/load_brats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/datasets/load_brats.py -------------------------------------------------------------------------------- /diff_scm/datasets/load_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/datasets/load_mnist.py -------------------------------------------------------------------------------- /diff_scm/datasets/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/datasets/loader.py -------------------------------------------------------------------------------- /diff_scm/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/models/__init__.py -------------------------------------------------------------------------------- /diff_scm/models/gaussian_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/models/gaussian_diffusion.py -------------------------------------------------------------------------------- /diff_scm/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/models/losses.py -------------------------------------------------------------------------------- /diff_scm/models/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/models/nn.py -------------------------------------------------------------------------------- /diff_scm/models/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/models/resample.py -------------------------------------------------------------------------------- /diff_scm/models/respace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/models/respace.py -------------------------------------------------------------------------------- /diff_scm/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/models/unet.py -------------------------------------------------------------------------------- /diff_scm/sampling/sample_counterfactual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/sampling/sample_counterfactual.py -------------------------------------------------------------------------------- /diff_scm/sampling/sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/sampling/sampling_utils.py -------------------------------------------------------------------------------- /diff_scm/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diff_scm/training/anticausal_classifier_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/training/anticausal_classifier_train.py -------------------------------------------------------------------------------- /diff_scm/training/main_diffusion_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/training/main_diffusion_train.py -------------------------------------------------------------------------------- /diff_scm/training/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/training/train_util.py -------------------------------------------------------------------------------- /diff_scm/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diff_scm/utils/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/utils/dist_util.py -------------------------------------------------------------------------------- /diff_scm/utils/fp16_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/utils/fp16_util.py -------------------------------------------------------------------------------- /diff_scm/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/utils/logger.py -------------------------------------------------------------------------------- /diff_scm/utils/script_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/utils/script_util.py -------------------------------------------------------------------------------- /diff_scm/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/diff_scm/utils/utils.py -------------------------------------------------------------------------------- /docs/_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/docs/_config.yml -------------------------------------------------------------------------------- /docs/assets/diffusion_causal_relationships.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/docs/assets/diffusion_causal_relationships.png -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/docs/index.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/environment.yml -------------------------------------------------------------------------------- /examples/example_generate_counterfactual.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vios-s/Diff-SCM/HEAD/examples/example_generate_counterfactual.ipynb --------------------------------------------------------------------------------