Add chain loader

This commit is contained in:
Robert Miles 2018-08-03 01:43:48 -04:00
parent 25f7075304
commit 4ea6c0a7f6
1 changed files with 19 additions and 0 deletions

19
chain.py Normal file
View File

@ -0,0 +1,19 @@
import markovify,os
# on by default, if it lags turn it off
RETAIN_ORIGINAL = True
combined_model = None
for corpus in os.listdir("corpora"):
with open(os.path.join("corpora",corpus)) as f:
model = markovify.Text(f.read(),retain_original=RETAIN_ORIGINAL)
if combined_model:
combined_model = markovify.combine(models=[combined_model,model])
else:
combined_model = model
def make_sentence(*args,**kwargs):
return combined_model.make_sentence(*args,**kwargs)
if __name__=="__main__":
print(combined_model.make_sentence(tries=100))