This commit is contained in:
Nick Walton
2019-11-14 17:10:21 -07:00
parent 22a1c95490
commit 37a7ccb336
2 changed files with 8 additions and 2 deletions
+1 -1
View File
@@ -18,7 +18,7 @@ tf.logging.set_verbosity(tf.logging.ERROR)
class SimpleGenerator:
def __init__(self, generate_num=40, temperature=0.9, top_k=40, top_p=0.0):
def __init__(self, generate_num=65, temperature=0.9, top_k=40, top_p=0.0):
self.generate_num=generate_num
self.temp = temperature
self.top_k = top_k
+7 -1
View File
@@ -41,7 +41,13 @@ class Story():
self.results.append(story_block)
def latest_result(self):
if len(self.results) >= 2:
if len(self.results) >= 4:
return self.context + self.actions[-4] + self.results[-4] + self.actions[-3] + self.results[-3] + \
self.actions[-2] + self.results[-2] + self.actions[-1] + self.results[-1]
elif len(self.results) >= 3:
return self.context + self.actions[-3] + self.results[-3] + self.actions[-2] + self.results[-2] + \
self.actions[-1] + self.results[-1]
elif len(self.results) >= 2:
return self.context + self.actions[-2] + self.results[-2] + self.actions[-1] + self.results[-1]
elif len(self.results) >= 1:
return self.context + self.actions[-1] + self.results[-1]