Skip to content

text-rnn allows you to create modern neural network architectures which use modern techniques such as skip-embedding and attention weighting. Train either a bidirectional or normal LSTM recurrent neural network to generate text using any dataset. You can continue training a pre-trained model.

License

Notifications You must be signed in to change notification settings

demmojo/text-rnn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

text-rnn

text-rnn allows you to create modern neural network architectures which use modern techniques such as skip-embedding and attention weighting. It trains and generate text at the character-level. It also uses the CuDNN implementation when trained on GPUs which significantly improves training time when compared to the usual implementation of LSTMs.

You can configure whether to use bidirectional RNNs, the number of RNN layers, RNN size, input length, and size of the embedding layer.

If you would like to train using a free GPU check out this Colaboratory notebook.

About

text-rnn allows you to create modern neural network architectures which use modern techniques such as skip-embedding and attention weighting. Train either a bidirectional or normal LSTM recurrent neural network to generate text using any dataset. You can continue training a pre-trained model.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages