/GAN-EM

GAN based EM learning framework for clustering, semi-supervised classification and dimensionality reduction.

Primary LanguagePython

GAN-EM

Implementation of paper "GAN-EM: GAN based EM Learning Framework".

Requirements:

  • python: tested with 2.7.13 & 3.6.4
  • PyTorch 0.4.0

Training for MNIST

  • Download MNIST dataset and store it under data/mnist/
  • Run main.py