From 0fe73ab7f6324b691d47cc362b14392a61359ade Mon Sep 17 00:00:00 2001 From: Nick Date: Fri, 20 Sep 2019 17:30:35 -0600 Subject: [PATCH] Update --- console_play.py | 5 ++++- generator/ctrl/ctrl_generator.py | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/console_play.py b/console_play.py index b3794df..d1ccb72 100644 --- a/console_play.py +++ b/console_play.py @@ -57,6 +57,9 @@ def play_constrained(): result = None while(result == None): action_choice = input("Which action do you choose? ") + if action_choice is "print story": + print(story_manager.story) + continue print("\n") result, possible_actions = story_manager.act(action_choice) @@ -84,7 +87,7 @@ def play_cached(): if __name__ == '__main__': - play_constrained() + play_unconstrained() diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index a646942..8aef60d 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -22,7 +22,7 @@ def loss(labels, logits): class CTRLGenerator(): - def __init__(self, control_code="Horror Text: ", generate_num=64, temperature=0.5): + def __init__(self, control_code="Horror Text: ", generate_num=64, temperature=0.75): self.generate_num=generate_num model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/"