Skip to content

Instantly share code, notes, and snippets.

@himanish-star

himanish-star/train_all_sentence_iterative.py

Last active March 19, 2020 14:13
Show Gist options
  • Save himanish-star/b2190c8a0ece03044a022557b761c5b5 to your computer and use it in GitHub Desktop.
Save himanish-star/b2190c8a0ece03044a022557b761c5b5 to your computer and use it in GitHub Desktop.
Download ZIP
script to train model on all sentences
Raw
train_all_sentence_iterative.py
from textgenrnn import textgenrnn
textgen = textgenrnn(weights_path="",vocab_path="",config_path="")
# textgen.generate()
textgen.train_from_file('part_1.txt', num_epochs=20, new_model=True)
textgen.save(weights_path="textgenrnn_all_sentences_weights_saved.hdf5")
textgen.generate()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment