Skip to content

Commit

Permalink
Merge 2d063e0 into fc11a66
Browse files Browse the repository at this point in the history
  • Loading branch information
LeonardoAlchieri committed Sep 23, 2020
2 parents fc11a66 + 2d063e0 commit 7c0cd08
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion responsibly/we/utils.py
Expand Up @@ -81,7 +81,10 @@ def cosine_similarities_by_words(model, word, words):

vec = model[word]
vecs = [model[w] for w in words]
return model.cosine_similarities(vec, vecs)
# This line was changed by Leonardo Alchieri
# It is done in order to function with later
# versions of gensim.
return model.wv.cosine_similarities(vec, vecs)


def update_word_vector(model, word, new_vector):
Expand Down

0 comments on commit 7c0cd08

Please sign in to comment.