diff --git a/data/build_training_data.py b/data/build_training_data.py index 1d3dd8d..97b7990 100644 --- a/data/build_training_data.py +++ b/data/build_training_data.py @@ -33,8 +33,7 @@ def get_stories(filename): output_file_path = "text_adventures.txt" with open(output_file_path, 'w') as output_file: - filenames = ["stories/story" + str(i) + ".json" for i in range(0,69)] - filenames += ["stories/apoc_seed" + str(i) + ".json" for i in range(1,5)] + filenames = ["stories/story" + str(i) + ".json" for i in range(0,41)] stories = [] for filename in filenames: stories += get_stories(filename) diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 40afc4a..9829f04 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -11,7 +11,7 @@ tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR) class GPT2Generator: - def __init__(self, generate_num=80, temperature=0.4, top_k=40, top_p=0.8): + def __init__(self, generate_num=80, temperature=0.3, top_k=40, top_p=0.8): self.generate_num=generate_num self.temp = temperature self.top_k = top_k diff --git a/play.py b/play.py index 1a9c9d9..b7015a9 100644 --- a/play.py +++ b/play.py @@ -32,7 +32,7 @@ def get_num_options(num): def select_game(): print("Which game would you like to play?") - options = ["zombies", "hospital", "peasant", "apocalypse", "classic", "knight", "necromancer", "scifi", "vague"] + options = ["zombies", "hospital", "peasant", "apocalypse", "classic", "knight", "necromancer", "vague"] for i, option in enumerate(options): console_print(str(i) + ") " + option + ": " + get_context(option)) print("\n")