python - 如何在gensim中打印出LDA主题中单词的完整分布？

Question

以下代码中的 lda.show_topics 模块只打印每个主题前 10 个词的分布，我如何打印出语料库中所有词的完整分布？

from gensim import corpora, models

documents = ["Human machine interface for lab abc computer applications",
"A survey of user opinion of computer system response time",
"The EPS user interface management system",
"System and human system engineering testing of EPS",
"Relation of user perceived response time to error measurement",
"The generation of random binary unordered trees",
"The intersection graph of paths in trees",
"Graph minors IV Widths of trees and well quasi ordering",
"Graph minors A survey"]

stoplist = set('for a of the and to in'.split())
texts = [[word for word in document.lower().split() if word not in stoplist]
         for document in documents]

dictionary = corpora.Dictionary(texts)
corpus = [dictionary.doc2bow(text) for text in texts]

lda = models.ldamodel.LdaModel(corpus_tfidf, id2word=dictionary, num_topics=2)

for i in lda.show_topics():
    print i

Answer 1

show_topics() 中有一个变量调用 topn，您可以在其中指定每个主题的单词分布中所需的前 N ​​个单词的数量。见http://radimrehurek.com/gensim/models/ldamodel.html

所以不是默认的 lda.show_topics()。您可以使用 len(dictionary) 获取每个主题的完整单词分布:

for i in lda.show_topics(topn=len(dictionary)):
    print i