The MNIST dataset was used to train a neural network having a single linear layer with SoftMax employed in the criterion function (Cross Entropy Loss) to classify handwritten digits in classes 0 to 9. The model yielded a 92% accuracy on the MNIST test dataset in 10 training epochs.
-
Notifications
You must be signed in to change notification settings - Fork 0
The MNIST dataset was used to train a neural network having a single linear layer with SoftMax employed in the criterion function (Cross Entropy Loss) to classify handwritten digits in classes 0 to 9. The model yielded a 92% accuracy on the MNIST test dataset in 10 training epochs.
sharmeen-k/Digit_Classification_Softmax
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
The MNIST dataset was used to train a neural network having a single linear layer with SoftMax employed in the criterion function (Cross Entropy Loss) to classify handwritten digits in classes 0 to 9. The model yielded a 92% accuracy on the MNIST test dataset in 10 training epochs.
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published