Skip to content

oxyboron25/mnist-digit-recognition

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 

Repository files navigation

MNIST Digit Recognition

This repository contains code for training and evaluating a digit recognition model using the MNIST dataset.

Requirements

  • Python 3.x
  • TensorFlow 2.x
  • NumPy
  • Matplotlib

Installation

Clone the repository:

git clone https://github.com/oxyboron25/mnist-digit-recognition.git
cd mnist-digit-recognition

Install dependencies using pip:

pip install -r requirements.txt

Usage

Training the Model

To train the model, run:

python train.py

This will train the digit recognition model using the MNIST dataset.

Evaluating the Model

To evaluate the model on the test set, run:

python evaluate.py

This will evaluate the accuracy of the trained model on the MNIST test dataset.

Predicting Digit Images

To predict digits from custom images, use:

python predict.py path_to_image

Replace path_to_image with the path to the image file you want to predict.

Structure

  • data/: Contains the MNIST dataset (automatically downloaded if not present).
  • models/: Saved models after training.
  • train.py: Script for training the digit recognition model.
  • evaluate.py: Script for evaluating the model's accuracy.
  • predict.py: Script for predicting digits from custom images.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published