Skip to content

Commit cc3a979

Browse files
Update README.md
1 parent 8d8fd3d commit cc3a979

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

README.md

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,24 @@
11
# MNIST Dataset Classification
2+
23
> ~A standard (non-convolution based) neural network to classify the MNIST dataset.
34
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+
422
The MNIST Database contains gray-scale images of 28x28 dimension where each image represents a handwritten digit which the network has to identify.
523

624
## Step 1 : Setting up the database
@@ -146,3 +164,5 @@ view_classify(image.view(1, 28, 28), ps)
146164
2. Looking to test with different algorithms to strike a balance between training time and accuracy.
147165

148166
### 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

Comments
 (0)