Skip to content

Rayan-Azrai/MNIST-model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

7 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

MNIST Classification Model

A fully-connected neural network trained on the MNIST handwritten-digit dataset using PyTorch.
This notebook demonstrates data loading, model definition, training, evaluation, and visualization of results.


๐Ÿš€ Features

  • Data Loading via torchvision.datasets.MNIST
  • Reproducible Training (fixed seeds for Python, NumPy, and PyTorch)
  • Simple Feed-Forward Classifier using nn.Sequential and nn.Flatten
  • Training & Validation Loop with loss tracking over epochs
  • Evaluation Metrics:
    • Classification report (precision, recall, F1-score)
    • Confusion matrix visualized with Seaborn
  • Device-Agnostic: runs on GPU if available, otherwise CPU

About

Neural networks model on the MNIST dataset using pytorch

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published