减少了一些训练轮次，重新验证了一下chatbot是否正常

qhduan · Apr 13, 2018 · cf37cab · cf37cab
1 parent a18aa08
commit cf37cab
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 7 deletions.
diff --git a/chatbot/params.json b/chatbot/params.json
@@ -6,7 +6,7 @@
     "cell_type": "lstm",
     "depth": 2,
     "attention_type": "Bahdanau",
-    "hidden_units": 1024,
+    "hidden_units": 128,
     "optimizer": "adam",
     "learning_rate": 0.001,
     "embedding_size": 300

diff --git a/chatbot/train.py b/chatbot/train.py
@@ -26,8 +26,8 @@ def test(params):
     ws = pickle.load(open('ws.pkl', 'rb'))
 
     # 训练部分
-    n_epoch = 5
-    batch_size = 32
+    n_epoch = 2
+    batch_size = 128
     # x_data, y_data = shuffle(x_data, y_data, random_state=0)
     # x_data = x_data[:10000]
     # y_data = y_data[:10000]

diff --git a/chatbot/train_anti.py b/chatbot/train_anti.py
@@ -26,8 +26,8 @@ def test(params):
     ws = pickle.load(open('ws.pkl', 'rb'))
 
     # 训练部分
-    n_epoch = 5
-    batch_size = 32
+    n_epoch = 2
+    batch_size = 128
     # x_data, y_data = shuffle(x_data, y_data, random_state=0)
     # x_data = x_data[:100000]
     # y_data = y_data[:100000]

diff --git a/chatbot_cut/train.py b/chatbot_cut/train.py
@@ -29,7 +29,7 @@ def test(bidirectional, cell_type, depth,
         open('chatbot.pkl', 'rb'))
 
     # 训练部分
-    n_epoch = 20
+    n_epoch = 5
     batch_size = 128
     # x_data, y_data = shuffle(x_data, y_data, random_state=0)
     # x_data = x_data[:10000]

diff --git a/chatbot_cut/train_anti.py b/chatbot_cut/train_anti.py
@@ -29,7 +29,7 @@ def test(bidirectional, cell_type, depth,
         open('chatbot.pkl', 'rb'))
 
     # 训练部分
-    n_epoch = 20
+    n_epoch = 5
     batch_size = 128
     # x_data, y_data = shuffle(x_data, y_data, random_state=0)
     # x_data = x_data[:100000]