revised twitter set for grulm

1 files changed, 7 insertions, 7 deletions

diff --git a/nerv/examples/lmptb/grulm_ptb_main.lua b/nerv/examples/lmptb/grulm_ptb_main.lua
index c5c011a..831146b 100644
--- a/nerv/examples/lmptb/grulm_ptb_main.lua
+++ b/nerv/examples/lmptb/grulm_ptb_main.lua
@@ -259,10 +259,10 @@ elseif (set == "twitter") then
 
 root_dir = '/home/slhome/txh18/workspace'
 data_dir = root_dir .. '/twitter_new/DATA'
-train_fn = data_dir .. '/twitter.choose.adds'
+train_fn = data_dir .. '/twitter.choose2.adds'
 valid_fn = data_dir .. '/twitter.valid.adds'
 test_fn = data_dir .. '/comm.test.choose-ppl.adds'
-vocab_fn = data_dir .. '/twitter.choose.train.vocab'
+vocab_fn = data_dir .. '/vocab.25000'
 
 --qdata_dir = root_dir .. '/ptb/questionGen/gen'
 
@@ -275,22 +275,22 @@ global_conf = {
     hidden_size = 300,
     layer_num = 1,
     chunk_size = 15,
-    batch_size = 20, 
-    max_iter = 35,
+    batch_size = 32, 
+    max_iter = 30,
     lr_decay = 1.003,
     decay_iter = 10,
     param_random = function() return (math.random() / 5 - 0.1) end,
-    dropout_str = "0",
+    dropout_str = "0.5",
 
     train_fn = train_fn,
     valid_fn = valid_fn,
     test_fn = test_fn,
     vocab_fn = vocab_fn,
-    max_sen_len = 90,
+    max_sen_len = 32,
     sche_log_pre = "[SCHEDULER]:",
     log_w_num = 40000, --give a message when log_w_num words have been processed
     timer = nerv.Timer(),
-    work_dir_base = root_dir .. '/twitter_new/EXP-nerv/lstmlm_v1.0'
+    work_dir_base = root_dir .. '/twitter_new/EXP-nerv/grulm_v1.0'
 }
 
 else