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號碼?