我可以获得LDA中单词的主题分布吗？

Question

我是LDA的新手，我想计算单词之间的主题相似度。我可以得到一个词的主题分布吗？如果是这样，我如何在 gensim.ldamodel 中执行此操作？

Answer 1

Gensim 的 LDA mallet wrapper 有一个 load_word_topics() 函数（我假设它的 python LDA 实现也是如此）。它 returns 是单词 X 主题的矩阵。从中，您可以获得每个主题中每个单词的频率向量。那将是给定词的主题词分布。