culled stories

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)

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

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")