digit-classifier / README.md
dcarpintero's picture
Update README.md
8248475 verified
|
raw
history blame
622 Bytes
metadata
license: apache-2.0

Implemented as a Multi-Layer Perceptron to classify handwritten Digits (0-9)

[Annotated Notebook]

Model Architecture and Results

The model comprises a flattening layer and three linear layers ((128, 64) hidden dimensions) with relus to approximate non-linearity. It achieves 90% accuracy after 10 training epochs and batch size = 128. Taining and Test MNIST datasets are loaded with PyTorch dataloaders.