DEPRECATED CODE : Text generation using RNN (LSTM) implemented using Tensorflow
Switch branches/tags
Nothing to show
Clone or download
spiglerg Merge pull request #13 from mscam/master
Fix session save on batch iterations.
Latest commit 7e6ec16 Feb 8, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data Initial commit Sep 2, 2016
.gitignore Code cleanup Dec 9, 2017
LICENSE Initial commit Sep 2, 2016
README.md mscam pull request merge Dec 14, 2017
requirements-dev.txt mscam pull request merge Dec 14, 2017
requirements.txt mscam pull request merge Dec 14, 2017
rnn_tf.py Fix session save on batch iterations. Jan 28, 2018
setup.cfg Code cleanup Dec 9, 2017

README.md

Text generation using a RNN

Text generation using a RNN (LSTM) using Tensorflow.

Usage

To train the model you can set the textfile you want to use to train the network by using command line options:

Run the network in train mode:

$ python rnn_tf.py --input_file=data/shakespeare.txt --ckpt_file="saved/model.ckpt" --mode=train

Run the network to generate text:

$ python rnn_tf.py --input_file=data/shakespeare.txt --ckpt_file="saved/model.ckpt" --test_prefix="The " --mode=talk