fix embedding text_one

yongzhuo · Feb 1, 2020 · 1178950 · 1178950
1 parent 636afc4
commit 1178950
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/macropodus/network/base/embedding.py b/macropodus/network/base/embedding.py
@@ -142,7 +142,7 @@ def deal_corpus(self):
                     else:
                         raise RuntimeError("your input level_type is wrong, it must be 'word', 'char', 'ngram'")
                     for text_one in text:
-                        if term_one not in token2idx:
+                        if text_one not in token2idx:
                             token2idx[text_one] = len(token2idx)
         else:
             raise RuntimeError("your input corpus_path is wrong, it must be 'dict' or 'corpus'")