2016-01-20 94 views
0

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

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

当doc2vec模型训练和推理使用版本0.12.3本代码给出的结果做了如下图所示:

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

当doc2vec模型训练和推理是使用的版本0.12.2做了同样的代码给出的结果就像这样:

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

我如何获得文件的标签,而不是在0.12.3号码?

回答