fix util #54

aonotas · Dec 13, 2017 · d5764df · d5764df
1 parent 96c3126
commit d5764df
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deepcrf/util.py b/deepcrf/util.py
@@ -62,7 +62,9 @@ def load_vocab(filename):
     vocab = {}
     with open(filename) as f:
         for l in f:
-            w, idx = str_to_unicode_python2(l).strip().split(u'\t')
+            lst = str_to_unicode_python2(l).strip().split(u'\t')
+            w = lst[0]
+            idx = lst[-1]
             vocab[w] = int(idx)
     return vocab