Update preprocess.py

harvardnlp · Jul 31, 2020 · b1eeffa · b1eeffa
1 parent 28917b3
commit b1eeffa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/preprocess.py b/preprocess.py
@@ -59,6 +59,7 @@ def prune_vocab(self, k, cnt = False):
 
     def load_vocab(self, vocab_file):
         self.d = {}
+        self.idx2word = {}
         for line in open(vocab_file, 'r'):
             v, k = line.strip().split()
             self.d[v] = int(k)