Tensorflow Demo
Overview
This project helps train a classifier to recognize handwritten character images MNIST digits. It uses logistic regression as it's model and trains on a small MNIST dataset before testing the trained model on it. You can view the constructed data flow graph using Tensorboard in your browser after training. This is the code for TensorFlow in 5 Min on Youtube
Dependencies
TensorFlow (https://www.tensorflow.org/versions/r0.10/get_started/os_setup.html#pip-installation)
Use pip to install any missing dependencies
Basic Usage
-
Change Line 56 in board.py to a location on your computer
-
Run
python board.pyto train the model. It will download & split the MNISt dataset into training and testing data. Then it will train a logistic regression model on the data. Lastly, it will test the trained model on the test set. -
You can visualize the model in tensorboard by running
tensorboard --logdir=LOCATION_ON_YOUR_COMPUTER
That's it!
Credits
Credit for the vast majority of code here goes to Google! I've merely created a wrapper around all of the important functions to get people started.