textgenrnn utility scrips

rnn.py

import sys, os, re
import pprint
from util import *
import re
from textgenrnn import textgenrnn

textgen = textgenrnn()

def gen(cnt="10", temp="1.0"):
	textgen.generate(int(cnt), temperature=float(temp))

if __name__ == "__main__":
	textgen.load(sys.argv[1])
	gen(sys.argv[2], sys.argv[3])

train.py

import sys
from textgenrnn import textgenrnn

def train(s, iterations="20"):
	textgen = textgenrnn()
	textgen.train_from_file(s, num_epochs=int(iterations))
	textgen.save(s+'.hdf5')
	textgen.generate(10)

if __name__ == "__main__":
	train(sys.argv[1], sys.argv[2])

usage.md

train a .hdf5 file, iteration number helps things a little bit, but it's pretrained so will work with 1

train.py mysource.txt 20

create output, the 3rd arg is the 'temperature', higher temps deviate i.e. give more nonsense

rnn.py seinfeld.hdf5 1000 0.7 > seinfeld.out.txt

Thank you!