train_textGenRNN.py

from textgenrnn import textgenrnn 
import glob

textgen = textgenrnn()
# textGenRNN.train_from_file requires the whole dataset to be in a single file

corpus = ''
for file_name in glob.glob('lovecraft_corpus/*'):
    with open(file_name) as f:
        corpus+=f.read()
with open('lovecraft_corpus.txt', 'w') as f:
    f.write(corpus)
    
textgen.train_from_file('lovecraft_corpus.txt', num_epochs = 1, batch_size = 4096)