一段时间以来,我一直在试验 doc2vec 模块。我可以训练我的模型并让训练后的模型为给定文档输出类似的文档,如下所示:
import re
modelloaded=Doc2Vec.load("model_all_doc_dm_1")
st = 'long description of a document as string'
doc = re.sub('[^a-zA-Z]', ' ', st).lower().split()
new_doc_vec = modelloaded.infer_vector(doc)
modelloaded.docvecs.most_similar([new_doc_vec])
这很好用,给了我 10 个结果。有没有办法获得超过 10 个结果或者这是限制?
最佳答案
我找到了:
modelloaded.docvecs.most_similar([new_doc_vec], topn=N)
topn=N
句柄可用于获得 10 多个结果。
关于python-3.x - Gensim 的 Doc2Vec most_similar 文档结果集中有限制吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33789541/