使用word2vec计算句子相似度

时间:2017-01-12 21:18:16

标签: python vector word2vec cosine-similarity sentence

在上一篇文章中,我找到了一些描述计算2个句子之间语义相似度的方法的代码。

我的问题是我需要什么库,模块等(例如来自NAME导入NAME)才能在我的计算机上运行此代码。 Link to code

我在想也许是word2vec,numpy,scikit学习,但我不确定。

2 个答案:

答案 0 :(得分:2)

基本上你需要的是:

pretrained word vector
gensim
numpy
scipy 

对于语义,您需要单词向量,以便您可以计算句子之间的相似性。

以下是一步一步的教程:How to calculate phrase similarity between phrases

答案 1 :(得分:0)

您需要在代码中添加下一个导入:

from scipy import spatial
from gensim import models
import numpy as np

如果您使用的是 Anaconda Distribution ,则可以使用以下命令安装gensim:

conda install -c anaconda gensim=0.13.3

请记住,为了制作可运行的代码,您需要一些模型