fix util #54

aonotas · Dec 13, 2017 · fdfcbd3 · fdfcbd3
1 parent 10c59ee
commit fdfcbd3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepcrf/util.py b/deepcrf/util.py
@@ -63,7 +63,7 @@ def load_vocab(filename):
     with open(filename) as f:
         for l in f:
             lst = str_to_unicode_python2(l).strip().split(u'\t')
-            w = lst[0]
+            w = u'\t'.join(lst[:-1])
             idx = lst[-1]
             vocab[w] = int(idx)
     return vocab