Merge pull request #401 from frankwhzhang/fix_w2c_0323

fix word2vec reader
PaddlePaddle · Mar 23, 2021 · 758799e · 758799e
2 parents 4a6006e + d85651e
commit 758799e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/recall/word2vec/word2vec_reader.py b/models/recall/word2vec/word2vec_reader.py
@@ -94,8 +94,8 @@ def __iter__(self):
                     for idx, target_id in enumerate(word_ids):
                         context_word_ids = self.get_context_words(word_ids,
                                                                   idx)
-                        output = []
                         for context_id in context_word_ids:
+                            output = []
                             output.append(
                                 np.array([int(target_id)]).astype('int64'))
                             output.append(