- Python 3.5 +
- Scikit-Learn (latest version)
- Numpy (+ mkl for Windows)
- Download the four MNIST dataset files from this link:
- Unzip and place the files in the dataset folder inside the MNIST_Dataset_Loader folder under each ML Algorithm folder i.e :
KNN |_ MNIST_Dataset_Loader |_ dataset |_ train-images-idx3-ubyte |_ train-labels-idx1-ubyte |_ t10k-images-idx3-ubyte |_ t10k-labels-idx1-ubyte
Do this for SVM and RFC folders and you should be good to go.
- To run the code, navigate to one of the directories for which you want to run the code using command prompt:
cd 1. K Nearest Neighbors/
and then run the file "knn.py" as follows:
This will run the code and all the print statements will be logged into the "summary.log" file.
NOTE: If you want to see the output to print on the Command prompt, just comment out line 16, 17, 18, 106 and 107 and hence you will get all the prints on the screen.
Alternatively, you can also use PyCharm to run the code and run the ".py" file in there.
Repeat the steps for SVM and RFC code.
- To run the CNN code, you don't need to provide in the MNIST dataset as it'll be downloaded automatically.
Just run the file as :
and it should run fine.
- If you want to save the CNN model weights after training, run the code with the following arguments:
python CNN_MNIST.py --save_model 1 --save_weights cnn_weights.hdf5
python3 CNN_MNIST.py --save_model 1 --save_weights cnn_weights.hdf5
and it should save the model weights in the same directory.
- To load the saved model weights and avoid the training time again, use the following command:
python CNN_MNIST.py --load_model 1 --save_weights cnn_weights.hdf5
python3 CNN_MNIST.py --load_model 1 --save_weights cnn_weights.hdf5
and it should load the model and show the Outputs.
Accuracy using Machine Learning Algorithms:
K Nearest Neighbors: 96.67%
Random Forest Classifier: 96.82%
Accuracy using Deep Neural Networks:
Three Layer Convolutional Neural Network using Tensorflow: 99.70%
Three Layer Convolutional Neural Network using Keras and Theano: 98.75%
All code written in Python 3.5. Code executed on Intel Xeon Processor / AWS EC2 Server.