• Stars
    star
    111
  • Rank 314,510 (Top 7 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created about 8 years ago
  • Updated almost 7 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

DEPRECATED CODE : Text generation using RNN (LSTM) implemented using Tensorflow

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