fix(dawa): don't truncate corpus

parent 8de39959
......@@ -24,7 +24,7 @@ def train():
corpus = load_text()
print("Corpus:", corpus[:10])
lstm.create_model(corpus[:1000])
lstm.create_model(corpus)
with open(filename_output, "a+") as f:
for i in range(0, nb_epoch, 10):
lstm.fit(epochs=min(i + 10, nb_epoch), initial_epoch=i,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment