Skip to content

Commit f7f4f43

Browse files
committed
fix: make implicit conversion explicit
1 parent 6eaaaae commit f7f4f43

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

probabilistic_word_embeddings/evaluation.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -200,9 +200,9 @@ def words_in_e(row):
200200
r = len(df)
201201
target_words = list(df[columns[-1]])
202202

203-
X1 = embedding[df[columns[0]]]
204-
X2 = embedding[df[columns[1]]]
205-
X3 = embedding[df[columns[2]]]
203+
X1 = embedding[list(df[columns[0]])]
204+
X2 = embedding[list(df[columns[1]])]
205+
X3 = embedding[list(df[columns[2]])]
206206
X = X1 - X2 + X3
207207

208208
inv_vocab = {v: k for k, v in e.vocabulary.items()}

0 commit comments

Comments
 (0)