search

llSourcell / tensorflow_demo

Tensorflow Demo for my TF in 5 Min Video on Youtube
229 stars 174 forks source link

readme

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

  1. Change Line 56 in board.py to a location on your computer

  2. Run python board.py to 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.

  3. 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.