Merge pull request lisa-lab#116 from stray-leone/master

nouiz · nouiz · commit dc86d12f898d · 2015-12-15T13:11:50.000-05:00
change the way of getting vocsize, nclasses.
diff --git a/code/rnnslu.py b/code/rnnslu.py
@@ -288,10 +288,8 @@ def main(param=None):
     valid_lex, valid_ne, valid_y = valid_set
     test_lex, test_ne, test_y = test_set
 
-    vocsize = len(set(reduce(lambda x, y: list(x) + list(y),
-                             train_lex + valid_lex + test_lex)))
-    nclasses = len(set(reduce(lambda x, y: list(x)+list(y),
-                              train_y + test_y + valid_y)))
+    vocsize = len(dic['words2idx'])
+    nclasses = len(dic['labels2idx'])
     nsentences = len(train_lex)
 
     groundtruth_valid = [map(lambda x: idx2label[x], y) for y in valid_y]
