加载计算的 LDA 模型并打印每个主题最常见的单词

Question

我是 python 新手。我要做的是通过控制台读取 2 个参数。

1. 参数：使用 gensim 训练的 LDA 模型的路径。
2. 参数：我希望得到的每个主题最常用词的数量。

现在我想为所有主题打印每个主题最常见单词的数量。现在我的问题是如何获取所有主题。

这是我目前所拥有的：

import sys, getopt
import gensim

def main(argv):
   input_file = argv[0] #LDA Path
   number_of_words = argv[1] #Number of most common word per topic

   ldamodel = gensim.models.ldamodel.LdaModel.load(input_file, mmap=None) #load model
   ldamodel.print_topic(?, number_of_words)



if __name__ == "__main__":
   main(sys.argv[1:])

Gensim doc

谢谢

Answer 1

get_topic_terms 做你想做的事。我认为主题的数量存储在num_topics 变量中，可通过ldamodel.num_topics 访问。