jajokine / digit-recognizer

MITx - MicroMasters Program on Statistics and Data Science - Machine Learning with Python - Second Project

Digit-Recognizer

MITx - MicroMasters Program on Statistics and Data Science - Machine Learning with Python

Second Project - Non-Linear Classifier for Digit Analysis

The second project for the MIT MicroMasters Program course on Machine Learning with Python was aimed in discovering non-linear ways to classify images of digits using the famous MNIST (Mixed National Institute of Standards and Technology) database.

The MNIST database contains binary images of handwritten digits collected from among Census Bureau employees and high school students, and it is commonly used to train image processing systems. The database contains 60,000 training images and 10,000 testing images; All of which have been size-normalized and centered in a fixed size of 28 x 28 pixels.

The project consisted of first implementing linear methods, such as the closed form for linear regression and then Support Vector Machine with a linear decision boundary, before moving on towards non-linear methods with multinomial logistic regression (Softmax regression), and feature engineering with higher-order kernel methods, such as Polynomial and Radial Basis Function (RBF) kernels, as well as low-dimensional feature methods modular arithmetic (mod3) and Principal Components Analysis (PCA).

Dataset

The function call get_MNIST_data() returns the following Numpy arrays:

- train_x : A matrix of the training data. Each row of train_x contains the features of one image, which are
the raw pixel values flattened out into a vector of length 28^2 = 784. The pixel values are float values
between 0 and 1 (0 for black, 1 for white, and various shades of gray in-between.

- train_y : The labels for each training datapoint that are the digit numbers for each image (i.e. a number between 0-9).

- test_x : A matrix of the test data, formatted the same way as the training data.

- test_y : The labels for the test data, formatted the same way as the training data.

Access and requirements

The file main.py runs the code with the help of the six modules (features.py, kernel.py, linear_regression.py, softmax.py, svm.py and utils.py) that contain helper and utility functions, as well as the main functions for modeling and feature engineering.

The dependencies and requirements can be seen from requirements.txt that can be installed in shell with the command:

  pip install -r requirements.txt