Skip to content

Latest commit

 

History

History
29 lines (20 loc) · 764 Bytes

README.md

File metadata and controls

29 lines (20 loc) · 764 Bytes

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.