From 324aa6a559446f1f54ab387461f4518aa147c3d6 Mon Sep 17 00:00:00 2001
From: Nick <nickwalton00@gmail.com>
Date: Fri, 20 Sep 2019 13:12:41 -0600
Subject: [PATCH] update

---
 generator/ctrl/ctrl_generator.py | 2 +-
 story/utils.py                   | 1 +
 story/utils_test.py              | 0
 3 files changed, 2 insertions(+), 1 deletion(-)
 create mode 100644 story/utils_test.py

diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index 93c9646..ef33e8d 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -207,7 +207,7 @@ class CTRLGenerator():
             # Make sure only a possible verb is chosen.
             if first_token:
                 for word in get_possible_verbs():
-                    prompt_logits[_token][self.word2idx[word]] *= 1000
+                    prompt_logits[_token][self.word2idx[word]] += max(prompt_logits)
                 first_token = False
 
             # compute probabilities from logits
diff --git a/story/utils.py b/story/utils.py
index 0ab5b59..34d5618 100644
--- a/story/utils.py
+++ b/story/utils.py
@@ -102,6 +102,7 @@ second_to_first_mappings = [
     ("You are ", "I am "),
     ("You ", "I "),
     ("you ", "I "),
+    (" you.", " me."),
 ]
 
 def capitalize_helper(string):
diff --git a/story/utils_test.py b/story/utils_test.py
new file mode 100644
index 0000000..e69de29