A simple but powerful deep learning model that classifies handwritten digits (0β9) using the famous MNIST dataset.
This project is beginner-friendly and built using TensorFlow and Keras, achieving over 97% accuracy on test data. It's perfect for understanding how Artificial Neural Networks work.
- π 97%+ accuracy on test images
- π’ Trained on 60,000 handwritten digit images
- π§ Uses a basic Artificial Neural Network (ANN)
- πΌ Visualizes predictions vs real labels
- β‘ Built entirely in Google Colab
- π Easy to understand and modify
Tool | Purpose |
---|---|
Python | Programming language |
TensorFlow/Keras | Deep learning framework |
Matplotlib | Visualizing predictions |
Google Colab | Free GPU-powered training |
- Load the MNIST dataset (built into Keras)
- Normalize the data to improve training speed
- Build an ANN with:
- Input Layer: 28Γ28 image β Flattened
- Hidden Layer: 128 neurons with ReLU
- Output Layer: 10 neurons (softmax)
- Train the model over 5 epochs
- Evaluate performance on test data
- Visualize predictions
Prediction: 7
Actual Label: 7
Confidence:99.2%