├── .gitignore ├── README.md ├── configs ├── celeba.yml ├── cifar10.yml ├── cifar100.yml ├── fmnist.yml ├── mnist.yml ├── news20.yml └── omniglot.yml ├── main.py ├── minimal_requirements.txt ├── models ├── losses.py ├── model.py ├── model_smalltree.py └── networks.py ├── train ├── train.py ├── train_tree.py └── validate_tree.py ├── tree_exploration.ipynb ├── treevae.png ├── treevae.yml └── utils ├── data_utils.py ├── model_utils.py ├── plotting_utils.py ├── training_utils.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/README.md -------------------------------------------------------------------------------- /configs/celeba.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/configs/celeba.yml -------------------------------------------------------------------------------- /configs/cifar10.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/configs/cifar10.yml -------------------------------------------------------------------------------- /configs/cifar100.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/configs/cifar100.yml -------------------------------------------------------------------------------- /configs/fmnist.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/configs/fmnist.yml -------------------------------------------------------------------------------- /configs/mnist.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/configs/mnist.yml -------------------------------------------------------------------------------- /configs/news20.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/configs/news20.yml -------------------------------------------------------------------------------- /configs/omniglot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/configs/omniglot.yml -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/main.py -------------------------------------------------------------------------------- /minimal_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/minimal_requirements.txt -------------------------------------------------------------------------------- /models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/models/losses.py -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/models/model.py -------------------------------------------------------------------------------- /models/model_smalltree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/models/model_smalltree.py -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/models/networks.py -------------------------------------------------------------------------------- /train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/train/train.py -------------------------------------------------------------------------------- /train/train_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/train/train_tree.py -------------------------------------------------------------------------------- /train/validate_tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/train/validate_tree.py -------------------------------------------------------------------------------- /tree_exploration.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/tree_exploration.ipynb -------------------------------------------------------------------------------- /treevae.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/treevae.png -------------------------------------------------------------------------------- /treevae.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/treevae.yml -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/utils/model_utils.py -------------------------------------------------------------------------------- /utils/plotting_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/utils/plotting_utils.py -------------------------------------------------------------------------------- /utils/training_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/utils/training_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lauramanduchi/treevae/HEAD/utils/utils.py --------------------------------------------------------------------------------