gensim调用doc2vec计算句子的向量

在做句子的相似度时，会想到直接将句子表示成向量的形式，这样就可以将相似度计算的问题转换成两个向量之间的距离问题，网上找了一下发现，在gensim中的doc2vec可以实现将句子转换成向量，具体的代码如下所示：

# coding:utf-8
import jieba
import gensim
from gensim.models.doc2vec import Doc2Vec

TaggededDocument = gensim.models.doc2vec.TaggedDocument

def get_datasest():
    with open("linshi_file.txt", 'r') as cf: ##此处是获取你的训练集的地方，从一个文件中读取出来，里面的内容是一行一句话
        docs = cf.readlines()
    x_train = []
    for i, text in enumerate(docs):
        ##如果是已经分好词的，不用再进行分词，直接按空格切分即可
        word_list = ' '.join(jieba.cut(text.split('\n')[0])).encode('utf-8').split(' ')
        l = len(word_list)
        word_list[l - 1] = word_list[l - 1].strip()
        document = TaggededDocument(word_list, tags=[i])
        x_train.append(document)
    return x_train

def train(x_train, size=100, epoch_num=1): ##size 是你最终训练出的句子向量的维度，自己尝试着修改一下

    model_dm = Doc2Vec(x_train, min_count=1, window=5, size=size, sample=1e-3, negative=5, workers=4)
    model_dm.train(x_train, total_examples=model_dm.corpus_count, epochs=70)
    model_dm.save('model/model_dm_wangyi') ##模型保存的位置

    return model_dm


def ceshi():
    model_dm = Doc2Vec.load("model/model_dm_wangyi")
    ##此处需要读入你所需要进行提取出句子向量的文本   此处代码需要自己稍加修改一下
    str1 = '2018年03月23日晚上大概十一点多钟我和张三骑着摩托车从住处出门想看看有什么能吃的东西.'
    ##你需要进行得到句子向量的文本，如果是分好词的，则不需要再调用结巴分词
    test_text = ' '.join(jieba.cut(str1)).encode('utf-8').split(' ')

    inferred_vector_dm = model_dm.infer_vector(test_text) ##得到文本的向量
    print(inferred_vector_dm)

    return inferred_vector_dm


if __name__ == '__main__':
    x_train = get_datasest()
    model_dm = train(x_train)

    doc_2_vec = ceshi()
    print type(doc_2_vec)
    print(doc_2_vec.shape)

运行的结果显示如下所示：