From 208dd66808febaff8d4e60990dd7515ce78b3e4e Mon Sep 17 00:00:00 2001 From: Nick Date: Thu, 26 Sep 2019 10:17:39 -0600 Subject: [PATCH] refactored --- generator/ctrl/ctrl_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index a5fb09f..2377ca7 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -323,7 +323,7 @@ class CTRLGenerator(): if self.idx2word[idx] is "\n": num_new_lines += 1 - #print("Generated token is ", repr(self.idx2word[idx])) + print(self.idx2word[idx], end=" ") tokens_generated_so_far = ' '.join([self.idx2word[c] for c in tokens_generated[0][len(text):].squeeze()[:token + 2]]) tokens_generated_so_far = re.sub('(@@ )', '', string=tokens_generated_so_far)