Skip to content

Latest commit

 

History

History
 
 

mnist

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 

MNIST in TensorFlow

This directory builds a convolutional neural net to classify the MNIST dataset using the tf.contrib.data, tf.estimator.Estimator, and tf.layers APIs.

Setup

To begin, you'll simply need the latest version of TensorFlow installed.

First convert the MNIST data to TFRecord file format by running the following:

python convert_to_records.py

Then to train the model, run the following:

python mnist.py

The model will begin training and will automatically evaluate itself on the validation data.