yuhsinliu1993 / Expectation_Maximization

Expectation_Maximization

# Put MNIST data in 'mnist_data' folder and run the following command
$ python EM.py --dir='mnist_data' --num_classes=10

All options:

usage: EM.py [-h] [--dir DIR] [--num_classes NUM_CLASSES]

optional arguments:
  -h, --help            show this help message and exit
  --dir DIR             specify the data file
  --num_classes NUM_CLASSES
                        specify the number of classes