Skip to content

didw/tflearn_lstm_mnist

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Implemention LSTM example for MNIST data with tflearn

1. Tflearn is higher-level API for TensorFlow.

2. Dependencies: tensorflow 1.0, tflearn (sudo pip install tensorflow, tensorflow-gpu(GPU only), tflearn)

- Input: 28 rows of 28x1 fixel vector

- Output: 10 classes

3. Usage

- Training: python mnist_lstm.py

- Tensorboard: tensorboard --logdir=/tmp/tflearn_logs/

4. Result

$ python mnist_lstm.py

...
Training Step: 8600  | total loss: 0.03430 | time: 23.780s
| Adam | epoch: 010 | loss: 0.03430 - R2: 0.9763 -- iter: 55000/55000
training time: 131.180480003
predict time: 2.66158895493
accuracy:  97.53

5. Tensorboard

graph loss

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages