Add another print example

oreilly-japan · Jun 4, 2018 · 17faf16 · 17faf16
1 parent a62b8e0
commit 17faf16
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/ch07/peeky_seq2seq.py b/ch07/peeky_seq2seq.py
@@ -2,7 +2,7 @@
 import sys
 sys.path.append('..')
 from common.time_layers import *
-from ch07.seq2seq import Seq2seq, Encoder
+from seq2seq import Seq2seq, Encoder
 
 
 class PeekyDecoder:

diff --git a/ch07/rnnlm_gen.py b/ch07/rnnlm_gen.py
@@ -25,7 +25,7 @@ def generate(self, start_id, skip_ids=None, sample_size=100):
         return word_ids
 
     def get_state(self):
-        return (self.lstm_layer.h, self.lstm_layer.c)
+        return self.lstm_layer.h, self.lstm_layer.c
 
     def set_state(self, state):
         self.lstm_layer.set_state(*state)

diff --git a/ch07/show_addition_dataset.py b/ch07/show_addition_dataset.py
@@ -17,3 +17,8 @@
 print(t_train[0])
 # [ 3  0  2  0  0 11  5]
 # [ 6  0 11  7  5]
+
+print("".join([id_to_char[c] for c in x_train[0]]))
+print("".join([id_to_char[c] for c in t_train[0]]))
+# 71+118
+# _189