python - GloVe中两个词的相似度

Question

如何在 Glove 中找到两个单词的相似度？我们model.similarity(word1, word2)在 Word2vec 中有。GloVe中是否有类似的方法？如何保存和加载训练有素的 GloVe 模型？

Answer 1

你也可以使用scikit-learn

from sklearn.metrics.pairwise import cosine_similarity
cosine_similarity([[1, 0, -1]], [[-1,-1, 0]])
array([[-0.5]])

Answer 2

您可以使用余弦相似度。

编辑

您可以使用 SciPy 包。余弦相似度的 Python 代码：

from scipy import spatial

word_1 = [3, 45, 7, 2]
word_2 = [2, 54, 13, 15]
result = 1 - spatial.distance.cosine(word_1, word_2)