2016-01-20 91 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號碼?

回答