如何在 pyLDAvis 中获取每个主题的单词列表

Question

我是 pyLDAvis 的新手。我一直在浏览文档，但似乎无法找到一种方法来获取模型每个主题的词集。我有 20 个主题，我想为每个主题获取前 20 个左右的单词。有人有办法获取这些数据吗？

Answer 1

pyldavis.prepare() 方法生成一个 PreparedData 对象，其属性如 .topic_info returns 一个带有单词 logprob 等的 DataFrame（参见 docs)

from pyLDAvis.gensim import prepare
vis = prepare(lda_model, corpus, dictionary, mds='tsne')
vis.topic_info

     Category         Freq       Term        Total  loglift  logprob
term                                                                
2299  Default 2,068,609.00      order 2,068,609.00    30.00    30.00
1037  Default   816,951.00      drink   816,951.00    29.00    29.00
2778  Default   565,075.00     review   565,075.00    28.00    28.00

如何在 pyLDAvis 中获取每个主题的单词列表

How to get list of words for each topic in pyLDAvis

nlp

lda