SwissRoll VAE

Reconstruct Swiss roll using CSVAE and VAE

---

To run (VAE)

python3 vae_training.py

To run (CSVAE)

python3 training.py

---

1. Models
   • TEST_CLASS_ROLL - classification model ( swiss roll dataset)
   • TEST_VAE - vae model
     • results - recon_x_train.png ( reconstructed x_train )
     • results - recon_x_test.png ( reconstructed x_test )
     • results - intial_pca.png ( original x_train, x_test )
   • TEST_CSVAE - csvae model
     • results - recon_x_train.png ( reconstructed x_train )