Classify MNIST image dataset into 10 classes. Build an image classifier with Recurrent Neural Network (RNN: LSTM) on Tensorflow.
python3 train.py ./parameters.json
A directory will be created during training, and the model will be saved in this directory.
Provide the model directory (created when running train.py
) to predict.py
.
python3 predict.py ./trained_model_1481170507/