/mnist_cnn

Training a convolutional neural network on the MNIST dataset.

Primary LanguagePythonMIT LicenseMIT

MNIST Convolutional Neural Network

Overview

This code trains a convolutional neural network on the MNIST dataset. Uses the Keras and NumPy libraries.

Architecture

INPUT -> [CONV -> RELU -> BATCHNORM -> POOL]*2 -> FC -> RELU -> BATCHNORM -> DROPOUT -> FC

Accuracy

This model has achieved 99.28% test accuracy on the MNIST dataset with batch size of 128 and 12 epochs.

Dependencies

sudo pip3 install -r requirements.txt

Usage

python3 mnist_conv.py