├── .gitignore ├── README.md └── src ├── architectures ├── __init__.py ├── feedforward.py ├── helper.py ├── lstm_latent.py └── stochastic.py ├── commons ├── __init__.py ├── callbacks.py ├── func.py └── types_.py ├── datasets ├── __init__.py ├── data_modules.py ├── dsprites.py ├── helper.py └── mpi3d.py ├── evaluation ├── __init__.py ├── comp_gen.py ├── disentangle_metric_evaluator.py ├── extra_metrics_configs │ ├── beta_vae_sklearn.gin │ ├── dci.gin │ ├── factor_vae_metric.gin │ ├── irs.gin │ ├── mcc.gin │ ├── mig.gin │ ├── modularity_explicitness.gin │ └── sap_score.gin ├── group_metric.py ├── scikit_learn_evaluator.py ├── topo_sim.py └── utils.py ├── models ├── __init__.py ├── ae.py ├── beta_tcvae.py ├── optimizer.py ├── rec_el.py └── vae.py └── scripts ├── configs ├── rec_el.yaml ├── scikitlearn_eval.yaml └── vae.yaml ├── eval_gt_rep.py ├── experiments.py ├── run_el.py ├── run_tcvae.py └── run_vae.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/README.md -------------------------------------------------------------------------------- /src/architectures/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | 3 | -------------------------------------------------------------------------------- /src/architectures/feedforward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/architectures/feedforward.py -------------------------------------------------------------------------------- /src/architectures/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/architectures/helper.py -------------------------------------------------------------------------------- /src/architectures/lstm_latent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/architectures/lstm_latent.py -------------------------------------------------------------------------------- /src/architectures/stochastic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/architectures/stochastic.py -------------------------------------------------------------------------------- /src/commons/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/commons/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/commons/callbacks.py -------------------------------------------------------------------------------- /src/commons/func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/commons/func.py -------------------------------------------------------------------------------- /src/commons/types_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/commons/types_.py -------------------------------------------------------------------------------- /src/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/datasets/__init__.py -------------------------------------------------------------------------------- /src/datasets/data_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/datasets/data_modules.py -------------------------------------------------------------------------------- /src/datasets/dsprites.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/datasets/dsprites.py -------------------------------------------------------------------------------- /src/datasets/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/datasets/helper.py -------------------------------------------------------------------------------- /src/datasets/mpi3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/datasets/mpi3d.py -------------------------------------------------------------------------------- /src/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/__init__.py -------------------------------------------------------------------------------- /src/evaluation/comp_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/comp_gen.py -------------------------------------------------------------------------------- /src/evaluation/disentangle_metric_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/disentangle_metric_evaluator.py -------------------------------------------------------------------------------- /src/evaluation/extra_metrics_configs/beta_vae_sklearn.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/extra_metrics_configs/beta_vae_sklearn.gin -------------------------------------------------------------------------------- /src/evaluation/extra_metrics_configs/dci.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/extra_metrics_configs/dci.gin -------------------------------------------------------------------------------- /src/evaluation/extra_metrics_configs/factor_vae_metric.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/extra_metrics_configs/factor_vae_metric.gin -------------------------------------------------------------------------------- /src/evaluation/extra_metrics_configs/irs.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/extra_metrics_configs/irs.gin -------------------------------------------------------------------------------- /src/evaluation/extra_metrics_configs/mcc.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/extra_metrics_configs/mcc.gin -------------------------------------------------------------------------------- /src/evaluation/extra_metrics_configs/mig.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/extra_metrics_configs/mig.gin -------------------------------------------------------------------------------- /src/evaluation/extra_metrics_configs/modularity_explicitness.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/extra_metrics_configs/modularity_explicitness.gin -------------------------------------------------------------------------------- /src/evaluation/extra_metrics_configs/sap_score.gin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/extra_metrics_configs/sap_score.gin -------------------------------------------------------------------------------- /src/evaluation/group_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/group_metric.py -------------------------------------------------------------------------------- /src/evaluation/scikit_learn_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/scikit_learn_evaluator.py -------------------------------------------------------------------------------- /src/evaluation/topo_sim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/topo_sim.py -------------------------------------------------------------------------------- /src/evaluation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/evaluation/utils.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/models/__init__.py -------------------------------------------------------------------------------- /src/models/ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/models/ae.py -------------------------------------------------------------------------------- /src/models/beta_tcvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/models/beta_tcvae.py -------------------------------------------------------------------------------- /src/models/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/models/optimizer.py -------------------------------------------------------------------------------- /src/models/rec_el.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/models/rec_el.py -------------------------------------------------------------------------------- /src/models/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/models/vae.py -------------------------------------------------------------------------------- /src/scripts/configs/rec_el.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/scripts/configs/rec_el.yaml -------------------------------------------------------------------------------- /src/scripts/configs/scikitlearn_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/scripts/configs/scikitlearn_eval.yaml -------------------------------------------------------------------------------- /src/scripts/configs/vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/scripts/configs/vae.yaml -------------------------------------------------------------------------------- /src/scripts/eval_gt_rep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/scripts/eval_gt_rep.py -------------------------------------------------------------------------------- /src/scripts/experiments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/scripts/experiments.py -------------------------------------------------------------------------------- /src/scripts/run_el.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/scripts/run_el.py -------------------------------------------------------------------------------- /src/scripts/run_tcvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/scripts/run_tcvae.py -------------------------------------------------------------------------------- /src/scripts/run_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wildphoton/Compositional-Generalization/HEAD/src/scripts/run_vae.py --------------------------------------------------------------------------------