gensim计算词向量之间的距离
时间: 2024-02-09 15:10:46 浏览: 245
深度学习gensim训练词向量word2vec
可以使用gensim中的`similarity`方法来计算词向量之间的余弦相似度,然后通过余弦相似度计算距离。以下是一个示例代码:
```python
from gensim.models import Word2Vec
# 加载预训练好的模型
model = Word2Vec.load("path/to/pretrained/model")
# 计算两个词向量之间的距离
word1 = "apple"
word2 = "orange"
similarity = model.wv.similarity(word1, word2)
distance = 1 - similarity # 余弦相似度与距离成反比
print(distance)
```
你可以用上面的代码计算任意两个词向量之间的距离。注意,这里的距离指的是余弦相似度转换后的距离,即两个词向量之间的相似度。
阅读全文