/vi_vae_gmm

Variational Inference/Variational AutoEncoder + Gaussian Mixtures implementation in zhusuan

Primary LanguagePythonApache License 2.0Apache-2.0

Variational Inference/Variational AutoEncoder + Gaussian Mixtures implementation in zhusuan

  • vi_GMM_2d.py: a toy example of Variational Inference + Gaussian Mixture in 2D
  • gmvae.py: Variational AutoEncoder + Gaussian Mixture, using MNIST dataset