Skip to content

joycex99/hp-word-model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

hp-word-model

Word-level LSTM to train language models. The data directory provides the concatenatation of all seven Harry Potter books in a text file as input, but any piece of text can be used for the basis of the model.

To use, clone the directory, run train.py to train the model, and sample.py to generate arbitrarily long text samples.

The code also provides the option of using pretrained word embeddings such as word2vec or GloVe.

About

Probabilistic LSTM language model to generate Harry Potter texts. Implemented in TensorFlow

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages