Word Level LSTM model to Generate tv scripts
To train the Model
python tv-script-generator.py --data='./data/Friends_Script.txt'To generate the text
python generate.py --prime='Joey' --model='Model_Friends.pt' --length=400 -data='./data/Friends_Script.txt'Prime word is the first word of generated script, length is the no. of words in the generated script.
I have trained model for two tv-scripts:- ie Friends and Seinfeld.
Click here to download Friends Model
Click here to download Seinfeld Model
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.