Skip to content

This script trains a convolutional neural network (CNN) to classify handwritten digits.

Notifications You must be signed in to change notification settings

nmcintyre5/numberPredictions

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Convolutional Neural Network for MNIST Classification

Overview

This script trains a convolutional neural network (CNN) to classify handwritten digits from the MNIST dataset. The CNN architecture consists of multiple convolutional and pooling layers followed by fully connected layers. The model is trained using the Adam optimizer and categorical crossentropy loss function.

image of predictions

Key Features

  • Convolutional neural network (CNN) architecture
  • Training on the MNIST dataset
  • Evaluation using test set accuracy

image of model loss

  • Confusion matrix visualization

image of confusion matrix

How to Install

  1. Clone this repository to your local machine.
  2. Install the dependencies:
pip install numpy keras tensorflow scikit-learn matplotlib seaborn
  1. Run numberPredict.py
python numberPredict.py

Credits

  • The MNIST dataset was obtained here.
  • This code is inspired by Jousef Murad's Image Classification Model Project on Coursera.

Releases

No releases published

Packages

No packages published