我已将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中获取文档标签而不是数字?