We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 4a6006e + d85651e commit 758799eCopy full SHA for 758799e
models/recall/word2vec/word2vec_reader.py
@@ -94,8 +94,8 @@ def __iter__(self):
94
for idx, target_id in enumerate(word_ids):
95
context_word_ids = self.get_context_words(word_ids,
96
idx)
97
- output = []
98
for context_id in context_word_ids:
+ output = []
99
output.append(
100
np.array([int(target_id)]).astype('int64'))
101
0 commit comments