From 37a7ccb336d5f746f35a7d4894c9543d8bdc55a0 Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Thu, 14 Nov 2019 17:10:21 -0700 Subject: [PATCH] update --- generator/simple/simple_generator.py | 2 +- story/story_manager.py | 8 +++++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/generator/simple/simple_generator.py b/generator/simple/simple_generator.py index 0a4d78c..6db0667 100644 --- a/generator/simple/simple_generator.py +++ b/generator/simple/simple_generator.py @@ -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 diff --git a/story/story_manager.py b/story/story_manager.py index db60ae9..0381c64 100644 --- a/story/story_manager.py +++ b/story/story_manager.py @@ -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]