gensim中的Doc2vec推断0.12.3

时间:2016-01-20 10:55:03

标签: gensim

我已将gensim从0.12.2升级到0.12.3,我在doc2vec中进行推理时遇到了问题。这是执行推理的代码:

doc = query.lower().split()
inf_vec = formmodel.infer_vector(doc)
similarF = formmodel.docvecs.most_similar([inf_vec])

当使用版本0.12.3完成doc2vec模型训练和推理时,此代码给出的结果如下所示:

[(644539, 0.55715829133987427), (647249, 0.55713766813278198),...]

当使用版本0.12.2完成doc2vec模型训练和推理时,相同的代码给出了如下结果:

[(docId1, 0.55715829133987427), (docId2, 0.55713766813278198),....]

如何在0.12.3中获取文档标签而不是数字?

1 个答案:

答案 0 :(得分:0)

请查看此主题以获取解决方案gensim 0.12.3 inference