Skip to content

B
BabelZoo
Unverified Commit 8e6f7594 by PLN (Algolia)
Options

fix(dawa): don't truncate corpus

parent 8de39959
Showing with 1 addition and 1 deletion
KoozDawa/dawa.py
...@@ -24,7 +24,7 @@ def train(): ...@@ -24,7 +24,7 @@ def train():
corpus = load_text() corpus = load_text()
print("Corpus:", corpus[:10]) print("Corpus:", corpus[:10])
lstm.create_model(corpus[:1000]) lstm.create_model(corpus)
with open(filename_output, "a+") as f: with open(filename_output, "a+") as f:
for i in range(0, nb_epoch, 10): for i in range(0, nb_epoch, 10):
lstm.fit(epochs=min(i + 10, nb_epoch), initial_epoch=i, 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