vae.py contain a class to train a regular autoencoder, variational autoencoder (VAE) and importance weighted autoencoder (IWAE), vae_tensorflow.ipynb shows how to train the VAE and obtain latent space visualizations.
vae.py contain a class to train a regular autoencoder, variational autoencoder (VAE) and importance weighted autoencoder (IWAE), vae_tensorflow.ipynb shows how to train the VAE and obtain latent space visualizations.
MIT License