├── .gitignore ├── LICENSE ├── README.md ├── data_generation ├── README.md ├── data_generation_ball_in_boxes.py ├── data_generation_interventional_pong.py ├── data_generation_pinball.py ├── data_generation_voronoi.py └── temporal_causal3dident │ ├── data_generation_causal3dident.py │ ├── generate_causal3dident_images.py │ └── triplet_generation_causal3dident.py ├── environment.yml ├── experiments ├── README.md ├── datasets.py ├── train_ae.py ├── train_causal_encoder.py ├── train_nf.py ├── train_vae.py └── utils.py ├── figures ├── ball_in_boxes.png ├── citris_figure.png ├── icitris_setup.png ├── interventional_pong.gif ├── pinball.gif ├── temporal_causal3dident.png └── voronoi_benchmark.png └── models ├── ae ├── __init__.py └── lightning_module.py ├── baselines ├── __init__.py ├── ivae.py ├── slowvae.py └── utils.py ├── citris_nf ├── __init__.py └── lightning_module.py ├── citris_vae ├── __init__.py └── lightning_module.py ├── icitris_nf ├── __init__.py └── lightning_module.py ├── icitris_vae ├── __init__.py ├── lightning_module.py ├── mi_estimator.py ├── prior.py └── target_classifier.py └── shared ├── __init__.py ├── callbacks.py ├── causal_encoder.py ├── enco.py ├── encoder_decoder.py ├── flow_layers.py ├── modules.py ├── target_classifier.py ├── transition_prior.py ├── utils.py └── visualization.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/README.md -------------------------------------------------------------------------------- /data_generation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/data_generation/README.md -------------------------------------------------------------------------------- /data_generation/data_generation_ball_in_boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/data_generation/data_generation_ball_in_boxes.py -------------------------------------------------------------------------------- /data_generation/data_generation_interventional_pong.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/data_generation/data_generation_interventional_pong.py -------------------------------------------------------------------------------- /data_generation/data_generation_pinball.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/data_generation/data_generation_pinball.py -------------------------------------------------------------------------------- /data_generation/data_generation_voronoi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/data_generation/data_generation_voronoi.py -------------------------------------------------------------------------------- /data_generation/temporal_causal3dident/data_generation_causal3dident.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/data_generation/temporal_causal3dident/data_generation_causal3dident.py -------------------------------------------------------------------------------- /data_generation/temporal_causal3dident/generate_causal3dident_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/data_generation/temporal_causal3dident/generate_causal3dident_images.py -------------------------------------------------------------------------------- /data_generation/temporal_causal3dident/triplet_generation_causal3dident.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/data_generation/temporal_causal3dident/triplet_generation_causal3dident.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/environment.yml -------------------------------------------------------------------------------- /experiments/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/experiments/README.md -------------------------------------------------------------------------------- /experiments/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/experiments/datasets.py -------------------------------------------------------------------------------- /experiments/train_ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/experiments/train_ae.py -------------------------------------------------------------------------------- /experiments/train_causal_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/experiments/train_causal_encoder.py -------------------------------------------------------------------------------- /experiments/train_nf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/experiments/train_nf.py -------------------------------------------------------------------------------- /experiments/train_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/experiments/train_vae.py -------------------------------------------------------------------------------- /experiments/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/experiments/utils.py -------------------------------------------------------------------------------- /figures/ball_in_boxes.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/figures/ball_in_boxes.png -------------------------------------------------------------------------------- /figures/citris_figure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/figures/citris_figure.png -------------------------------------------------------------------------------- /figures/icitris_setup.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/figures/icitris_setup.png -------------------------------------------------------------------------------- /figures/interventional_pong.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/figures/interventional_pong.gif -------------------------------------------------------------------------------- /figures/pinball.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/figures/pinball.gif -------------------------------------------------------------------------------- /figures/temporal_causal3dident.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/figures/temporal_causal3dident.png -------------------------------------------------------------------------------- /figures/voronoi_benchmark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/figures/voronoi_benchmark.png -------------------------------------------------------------------------------- /models/ae/__init__.py: -------------------------------------------------------------------------------- 1 | from models.ae.lightning_module import Autoencoder -------------------------------------------------------------------------------- /models/ae/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/ae/lightning_module.py -------------------------------------------------------------------------------- /models/baselines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/baselines/__init__.py -------------------------------------------------------------------------------- /models/baselines/ivae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/baselines/ivae.py -------------------------------------------------------------------------------- /models/baselines/slowvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/baselines/slowvae.py -------------------------------------------------------------------------------- /models/baselines/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/baselines/utils.py -------------------------------------------------------------------------------- /models/citris_nf/__init__.py: -------------------------------------------------------------------------------- 1 | from models.citris_nf.lightning_module import CITRISNF -------------------------------------------------------------------------------- /models/citris_nf/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/citris_nf/lightning_module.py -------------------------------------------------------------------------------- /models/citris_vae/__init__.py: -------------------------------------------------------------------------------- 1 | from models.citris_vae.lightning_module import CITRISVAE -------------------------------------------------------------------------------- /models/citris_vae/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/citris_vae/lightning_module.py -------------------------------------------------------------------------------- /models/icitris_nf/__init__.py: -------------------------------------------------------------------------------- 1 | from models.icitris_nf.lightning_module import iCITRISNF -------------------------------------------------------------------------------- /models/icitris_nf/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/icitris_nf/lightning_module.py -------------------------------------------------------------------------------- /models/icitris_vae/__init__.py: -------------------------------------------------------------------------------- 1 | from models.icitris_vae.lightning_module import iCITRISVAE -------------------------------------------------------------------------------- /models/icitris_vae/lightning_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/icitris_vae/lightning_module.py -------------------------------------------------------------------------------- /models/icitris_vae/mi_estimator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/icitris_vae/mi_estimator.py -------------------------------------------------------------------------------- /models/icitris_vae/prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/icitris_vae/prior.py -------------------------------------------------------------------------------- /models/icitris_vae/target_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/icitris_vae/target_classifier.py -------------------------------------------------------------------------------- /models/shared/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/shared/__init__.py -------------------------------------------------------------------------------- /models/shared/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/shared/callbacks.py -------------------------------------------------------------------------------- /models/shared/causal_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/shared/causal_encoder.py -------------------------------------------------------------------------------- /models/shared/enco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/shared/enco.py -------------------------------------------------------------------------------- /models/shared/encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/shared/encoder_decoder.py -------------------------------------------------------------------------------- /models/shared/flow_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/shared/flow_layers.py -------------------------------------------------------------------------------- /models/shared/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/shared/modules.py -------------------------------------------------------------------------------- /models/shared/target_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/shared/target_classifier.py -------------------------------------------------------------------------------- /models/shared/transition_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/shared/transition_prior.py -------------------------------------------------------------------------------- /models/shared/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/shared/utils.py -------------------------------------------------------------------------------- /models/shared/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/phlippe/CITRIS/HEAD/models/shared/visualization.py --------------------------------------------------------------------------------