Skip to content

Commit

Permalink
Create README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
chinmayapancholi13 authored Mar 25, 2017
0 parents commit 47ba357
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# MNIST_CNN

The objective of this project is to implement a Convolutional Neural Network using Tensorflow for predicting MNIST images.

The MNIST dataset is downloaded from http://yann.lecun.com/exdb/mnist/ (four files) and extracted into a folder named 'data' just outside the folder containing the main.py file.
That is, the code reads the input data files from the folder '../data'. The train function in train_cnn.py trains the neural network given the training examples and saves the weights in a folder named 'weights_cnn' in the same folder as main.py.
The `test` function reads the saved weights and given the test examples it returns the predicted labels.

0 comments on commit 47ba357

Please sign in to comment.