Skip to content

Commit 758799e

Browse files
authored
Merge pull request #401 from frankwhzhang/fix_w2c_0323
fix word2vec reader
2 parents 4a6006e + d85651e commit 758799e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/recall/word2vec/word2vec_reader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,8 +94,8 @@ def __iter__(self):
9494
for idx, target_id in enumerate(word_ids):
9595
context_word_ids = self.get_context_words(word_ids,
9696
idx)
97-
output = []
9897
for context_id in context_word_ids:
98+
output = []
9999
output.append(
100100
np.array([int(target_id)]).astype('int64'))
101101
output.append(

0 commit comments

Comments
 (0)