Skip to content

PyTorch implementation of the Kaggle Digit Recognizer competition (MNIST). Includes data preprocessing, CNN model, training loop with early stopping, scheduler, and submission pipeline.

Notifications You must be signed in to change notification settings

snevil/mnist-digit-classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

mnist-digit-classifier

Kaggle Digit Recognizer (PyTorch)

This repository contains a clean and reproducible PyTorch implementation for the Kaggle Digit Recognizer competition.

The goal of the competition is to correctly classify handwritten digits (0–9) from the MNIST dataset.


Project Structure

  • data preparation → load CSVs, reshape/normalize images, build PyTorch datasets.
  • augmentation → random rotations, translations, normalization.
  • model → lightweight CNN with 3 convolutional layers + dropout.
  • training → training loop with validation, early stopping, and LR scheduler.
  • submission → generate submission.csv for Kaggle leaderboard.

⚙️ Setup

# clone the repository
git clone https://github.com/YOUR-USERNAME/digit-recognizer-pytorch.git
cd digit-recognizer-pytorch

# install dependencies
pip install torch torchvision pandas numpy matplotlib

About

PyTorch implementation of the Kaggle Digit Recognizer competition (MNIST). Includes data preprocessing, CNN model, training loop with early stopping, scheduler, and submission pipeline.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published