Counting 3,384 Big Data & Machine Learning Frameworks, Toolsets, and Examples...
Suggestion? Feedback? Tweet @stkim1

Author
Last Commit
Nov. 23, 2018
Created
Nov. 16, 2018

SOP-Generator

A simple LSTM based Statement of Purpose Generator for grad school. :)

Alt Text

Prerequisites

  • Pytorch 0.4
  • Python 2.x / 3.x

Generate

Generate samples from the trained model.

python generate.py --data ./data/sop/ --seed 6 # play with seed to get different text

Training

python main.py --cuda --epochs 500 --data data/sop/ --tied # Train a tied LSTM on SOP dataset with CUDA for 500 epochs

The model uses the nn.RNN module (and its sister modules nn.GRU and nn.LSTM) which will automatically use the cuDNN backend if run on CUDA with cuDNN installed.

During training, if a keyboard interrupt (Ctrl-C) is received, training is stopped and the current model is evaluated against the test dataset.

Contributors

Credits

pytorch/examples/tree/master/word_language_model