For dependencies, create a conda environment with: conda env create -f environment.yml
the vae_2d and vae_10 notebooks train and save the results of a vae with respectively 2 dimensions and 10 dimensions in the latent space.
the 2d and 10d files contain results and pickles of the trained models.