From f76d36a3452fa5526f4e254cdae3607f33cf6137 Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Fri, 27 Sep 2019 12:53:20 -0600 Subject: [PATCH] update --- console_play.py | 2 +- generator/ctrl/ctrl_generator.py | 2 +- story/story_data.yaml | 2 ++ 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/console_play.py b/console_play.py index 4deccf5..d6f7999 100644 --- a/console_play.py +++ b/console_play.py @@ -23,7 +23,7 @@ def console_print(str, pycharm=False): def play_unconstrained(): generator = CTRLGenerator() #generator = WebGenerator(CRED_FILE) - prompt = get_story_start("hospital") + prompt = get_story_start("vague_police") story_manager = UnconstrainedStoryManager(generator) story_manager.start_new_story(prompt) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 3935844..5405634 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -20,7 +20,7 @@ def loss(labels, logits): class CTRLGenerator(): - def __init__(self, control_code="Writing Text: ", generate_num=80, temperature=0.4): + def __init__(self, control_code="Writing Text: ", generate_num=80, temperature=0.7): self.generate_num=generate_num model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/" diff --git a/story/story_data.yaml b/story/story_data.yaml index c5329ee..bc2798a 100644 --- a/story/story_data.yaml +++ b/story/story_data.yaml @@ -5,6 +5,8 @@ prompts: forest: "You wake up in an old dark forest with no memory of how you got there. You look around and see " + vague_police: "You are a police officer. You get to work and " + action_verbs: classic: ["You attack", "You tell", "You use", "You go"]