From f18f99b606963b36fb17615891c1c330774a1eee Mon Sep 17 00:00:00 2001
From: Nick <nickwalton00@gmail.com>
Date: Thu, 19 Sep 2019 11:07:13 -0600
Subject: [PATCH] temperature

---
 generator/ctrl/ctrl_generator.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index 6a53c2d..c2cda8a 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -20,9 +20,9 @@ def loss(labels, logits):
 
 class CTRLGenerator():
 
-    def __init__(self, control_code="Horror Text: "):
+    def __init__(self, control_code="Horror Text: ", generate_num=128, temperature=0.3):
 
-        self.generate_num=256
+        self.generate_num=generate_num
         model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/"
         self.control_code = control_code
         vocab_file = 'generator/ctrl/model/vocab'
@@ -131,7 +131,7 @@ class CTRLGenerator():
         # load BPE codes
         self.bpe = fastBPE.fastBPE(code_file, vocab_file)
 
-        self.temperature = 0
+        self.temperature=temperature
         self.nucleusprob = 0
         self.penalty = 1.2
         self.topk = 0