|
1 | 1 | # MNIST Dataset Classification
|
| 2 | + |
2 | 3 | > ~A standard (non-convolution based) neural network to classify the MNIST dataset.
|
3 | 4 |
|
| 5 | +# Table of contents |
| 6 | + |
| 7 | +- [Step 1 : Setting up the database](#step-1--setting-up-the-database) |
| 8 | + - [Downloading and Transforming the database :](#downloading-and-transforming-the-database-) |
| 9 | + - [Getting to know the dataset better :](#getting-to-know-the-dataset-better-) |
| 10 | + - [Deciding on whether to use batches or not :](#deciding-on-whether-to-use-batches-or-not-) |
| 11 | +- [Step 2 : Creating the neural network](#step-2--creating-the-neural-network) |
| 12 | + - [Deciding on Number of Hidden Layers and neurons :](#deciding-on-number-of-hidden-layers-and-neurons-) |
| 13 | + - [Creating the Neural network Sequence :](#creating-the-neural-network-sequence-) |
| 14 | +- [Step 3 : Training the model on the dataset](#step-3--training-the-model-on-the-dataset) |
| 15 | +- [Step 4 : Testing the Model](#step-4--testing-the-model) |
| 16 | +- [Step 5 : Saving the model](#step-5--saving-the-model) |
| 17 | +- [To View results for any random picture in the dataset, the following code can be used :](#to-view-results-for-any-random-picture-in-the-dataset-the-following-code-can-be-used-) |
| 18 | + - [Examples](#examples-) |
| 19 | + - [Model Accuracy](#model-accuracy--the-accuracy-of-the-model-with-this-code-is-approximately-978-to-9802-with-a-training-time-of-aprox-35-to-4-minutes) |
| 20 | +- [Further Improvements](#further-improvements-) |
| 21 | + |
4 | 22 | The MNIST Database contains gray-scale images of 28x28 dimension where each image represents a handwritten digit which the network has to identify.
|
5 | 23 |
|
6 | 24 | ## Step 1 : Setting up the database
|
@@ -146,3 +164,5 @@ view_classify(image.view(1, 28, 28), ps)
|
146 | 164 | 2. Looking to test with different algorithms to strike a balance between training time and accuracy.
|
147 | 165 |
|
148 | 166 | ### Contributions, Suggestions, and inputs on graphical representation for better understanding are welcome.
|
| 167 | + |
| 168 | +# One of the trained model is uploaded to this repository as well for reference purposes. |
0 commit comments