From 5b4fa0fb538e4234e1658625513a2695a439d623 Mon Sep 17 00:00:00 2001 From: Nick Date: Tue, 24 Sep 2019 13:04:40 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 8 ++++++-- story/utils.py | 4 +++- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 25eda3b..85a227d 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -147,7 +147,8 @@ class CTRLGenerator(): # Disallow used verbs if "used_verbs" in options: for verb in options["used_verbs"]: - probabilities[self.word2idx[verb]] = -1e8 + if verb in self.word2idx: + probabilities[self.word2idx[verb]] = -1e8 return probabilities @@ -156,6 +157,9 @@ class CTRLGenerator(): if options is None: options = {} + if prompt[-2:] == ".I": + prompt = prompt[:-1] + " I" + if prompt[-1] != " ": prompt = prompt + " " first_token = True @@ -182,7 +186,7 @@ class CTRLGenerator(): padded_text = text + [0] * (total_text_len - len(text)) tokens_generated = np.tile(padded_text, (1, 1)) result = "" - max_new_lines = 5 + max_new_lines = 0 num_new_lines = 0 for token in range(len(text) - 1, total_text_len - 1): diff --git a/story/utils.py b/story/utils.py index 68fb7ab..994c123 100644 --- a/story/utils.py +++ b/story/utils.py @@ -143,7 +143,9 @@ def second_to_first_person(text): return capitalize_first_letters(text) -possible_verbs = ["ask", "go", "run", "open", "look", "walk", "make", "try", "say", "tell", "attack", "use", "turn", "fight", "scream", "yell"] +possible_verbs = ["get", "take", "put", "give","set", "keep", "help", "show", "pay", "read", "start", "stay", "call", + "change", "ask", "go", "run", "open", "look", "walk", "make", "try", "say", "tell", "attack", "use", + "turn", "fight", "scream", "yell"] def get_possible_verbs(): return possible_verbs \ No newline at end of file