From f2e15d6600ccda0db32eee98b2ca67c322071964 Mon Sep 17 00:00:00 2001
From: Coulson Zhang <719579371@qq.com>
Date: Thu, 6 Jun 2019 10:41:30 +0900
Subject: [PATCH] add some annotations

---
 main.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index dbc1248..ec3eff2 100644
--- a/main.py
+++ b/main.py
@@ -182,15 +182,15 @@ def main(datasets, U, n_epochs=20, batch_size=20, max_l=100, hidden_size=100, \
     rng = np.random.RandomState(r_seed)
     lsize, rsize = max_l, max_l
     sessionmask = T.matrix()
-    lx = []
-    lxmask = []
+    lx = []  #tokens from previous turns
+    lxmask = [] #masks from previous turns
     for i in range(max_turn):
         lx.append(T.matrix())
         lxmask.append(T.matrix())
 
     index = T.lscalar()
-    rx = T.matrix('rx')
-    rxmask = T.matrix()
+    rx = T.matrix('rx') #tokens from response
+    rxmask = T.matrix() #masks from response
     y = T.ivector('y')
     Words = theano.shared(value=U, name="Words")
     llayer0_input = []