├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── config ├── causalcircuit_betavae.yaml ├── causalcircuit_dvae.yaml ├── causalcircuit_ilcm.yaml ├── causalcircuit_reduce_dim.yaml ├── data │ ├── causalcircuit_raw.yaml │ ├── causalcircuit_reduced_dim.yaml │ └── scaling.yaml ├── eval │ └── default.yaml ├── general │ └── local.yaml ├── hydra.yaml ├── model │ ├── causalcircuit_betavae.yaml │ ├── causalcircuit_dvae.yaml │ ├── causalcircuit_ilcm.yaml │ ├── causalcircuit_reduce_dim.yaml │ ├── scaling_betavae.yaml │ ├── scaling_dvae.yaml │ └── scaling_ilcm.yaml ├── plotting │ └── default.yaml ├── scaling_betavae.yaml ├── scaling_dvae.yaml ├── scaling_ilcm.yaml └── training │ ├── causalcircuit_baselines.yaml │ ├── causalcircuit_fast.yaml │ ├── causalcircuit_ilcm.yaml │ ├── causalcircuit_reduce_dim.yaml │ ├── causalcircuit_reduce_dim_fast.yaml │ ├── scaling.yaml │ └── scaling_fast.yaml ├── experiments ├── __init__.py ├── causalcircuit.py ├── causalcircuit_reduce_dim.py ├── experiment_utils.py └── scaling.py ├── setup.py └── ws_crl ├── __init__.py ├── causal ├── __init__.py ├── graph.py ├── implicit_scm.py ├── interventions.py └── scm.py ├── distributions.py ├── encoder ├── __init__.py ├── base.py ├── flow.py ├── image_vae.py └── vae.py ├── gumbel.py ├── lcm ├── __init__.py ├── base.py ├── elcm.py ├── flow_lcm.py └── ilcm.py ├── metrics.py ├── nets.py ├── plotting.py ├── posthoc_graph_learning ├── __init__.py ├── enco.py └── implicit.py ├── splines.py ├── training.py ├── transforms.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/README.md -------------------------------------------------------------------------------- /config/causalcircuit_betavae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/causalcircuit_betavae.yaml -------------------------------------------------------------------------------- /config/causalcircuit_dvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/causalcircuit_dvae.yaml -------------------------------------------------------------------------------- /config/causalcircuit_ilcm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/causalcircuit_ilcm.yaml -------------------------------------------------------------------------------- /config/causalcircuit_reduce_dim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/causalcircuit_reduce_dim.yaml -------------------------------------------------------------------------------- /config/data/causalcircuit_raw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/data/causalcircuit_raw.yaml -------------------------------------------------------------------------------- /config/data/causalcircuit_reduced_dim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/data/causalcircuit_reduced_dim.yaml -------------------------------------------------------------------------------- /config/data/scaling.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/data/scaling.yaml -------------------------------------------------------------------------------- /config/eval/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/eval/default.yaml -------------------------------------------------------------------------------- /config/general/local.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/general/local.yaml -------------------------------------------------------------------------------- /config/hydra.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/hydra.yaml -------------------------------------------------------------------------------- /config/model/causalcircuit_betavae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/model/causalcircuit_betavae.yaml -------------------------------------------------------------------------------- /config/model/causalcircuit_dvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/model/causalcircuit_dvae.yaml -------------------------------------------------------------------------------- /config/model/causalcircuit_ilcm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/model/causalcircuit_ilcm.yaml -------------------------------------------------------------------------------- /config/model/causalcircuit_reduce_dim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/model/causalcircuit_reduce_dim.yaml -------------------------------------------------------------------------------- /config/model/scaling_betavae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/model/scaling_betavae.yaml -------------------------------------------------------------------------------- /config/model/scaling_dvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/model/scaling_dvae.yaml -------------------------------------------------------------------------------- /config/model/scaling_ilcm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/model/scaling_ilcm.yaml -------------------------------------------------------------------------------- /config/plotting/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/plotting/default.yaml -------------------------------------------------------------------------------- /config/scaling_betavae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/scaling_betavae.yaml -------------------------------------------------------------------------------- /config/scaling_dvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/scaling_dvae.yaml -------------------------------------------------------------------------------- /config/scaling_ilcm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/scaling_ilcm.yaml -------------------------------------------------------------------------------- /config/training/causalcircuit_baselines.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/training/causalcircuit_baselines.yaml -------------------------------------------------------------------------------- /config/training/causalcircuit_fast.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/training/causalcircuit_fast.yaml -------------------------------------------------------------------------------- /config/training/causalcircuit_ilcm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/training/causalcircuit_ilcm.yaml -------------------------------------------------------------------------------- /config/training/causalcircuit_reduce_dim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/training/causalcircuit_reduce_dim.yaml -------------------------------------------------------------------------------- /config/training/causalcircuit_reduce_dim_fast.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/training/causalcircuit_reduce_dim_fast.yaml -------------------------------------------------------------------------------- /config/training/scaling.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/training/scaling.yaml -------------------------------------------------------------------------------- /config/training/scaling_fast.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/config/training/scaling_fast.yaml -------------------------------------------------------------------------------- /experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/causalcircuit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/experiments/causalcircuit.py -------------------------------------------------------------------------------- /experiments/causalcircuit_reduce_dim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/experiments/causalcircuit_reduce_dim.py -------------------------------------------------------------------------------- /experiments/experiment_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/experiments/experiment_utils.py -------------------------------------------------------------------------------- /experiments/scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/experiments/scaling.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/setup.py -------------------------------------------------------------------------------- /ws_crl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ws_crl/causal/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/causal/__init__.py -------------------------------------------------------------------------------- /ws_crl/causal/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/causal/graph.py -------------------------------------------------------------------------------- /ws_crl/causal/implicit_scm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/causal/implicit_scm.py -------------------------------------------------------------------------------- /ws_crl/causal/interventions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/causal/interventions.py -------------------------------------------------------------------------------- /ws_crl/causal/scm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/causal/scm.py -------------------------------------------------------------------------------- /ws_crl/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/distributions.py -------------------------------------------------------------------------------- /ws_crl/encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/encoder/__init__.py -------------------------------------------------------------------------------- /ws_crl/encoder/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/encoder/base.py -------------------------------------------------------------------------------- /ws_crl/encoder/flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/encoder/flow.py -------------------------------------------------------------------------------- /ws_crl/encoder/image_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/encoder/image_vae.py -------------------------------------------------------------------------------- /ws_crl/encoder/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/encoder/vae.py -------------------------------------------------------------------------------- /ws_crl/gumbel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/gumbel.py -------------------------------------------------------------------------------- /ws_crl/lcm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/lcm/__init__.py -------------------------------------------------------------------------------- /ws_crl/lcm/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/lcm/base.py -------------------------------------------------------------------------------- /ws_crl/lcm/elcm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/lcm/elcm.py -------------------------------------------------------------------------------- /ws_crl/lcm/flow_lcm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/lcm/flow_lcm.py -------------------------------------------------------------------------------- /ws_crl/lcm/ilcm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/lcm/ilcm.py -------------------------------------------------------------------------------- /ws_crl/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/metrics.py -------------------------------------------------------------------------------- /ws_crl/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/nets.py -------------------------------------------------------------------------------- /ws_crl/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/plotting.py -------------------------------------------------------------------------------- /ws_crl/posthoc_graph_learning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/posthoc_graph_learning/__init__.py -------------------------------------------------------------------------------- /ws_crl/posthoc_graph_learning/enco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/posthoc_graph_learning/enco.py -------------------------------------------------------------------------------- /ws_crl/posthoc_graph_learning/implicit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/posthoc_graph_learning/implicit.py -------------------------------------------------------------------------------- /ws_crl/splines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/splines.py -------------------------------------------------------------------------------- /ws_crl/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/training.py -------------------------------------------------------------------------------- /ws_crl/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/transforms.py -------------------------------------------------------------------------------- /ws_crl/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Qualcomm-AI-research/weakly-supervised-causal-representation-learning/HEAD/ws_crl/utils.py --------------------------------------------------------------------------------