主題建模：LDA 模型中每個主題的 WordCloud

Question

問題：如何為 LDA 模型計算的每個主題創建一個詞雲。 我嘗試了以下方法，但似乎無法進一步為每個主題創建一個詞雲。

first_topic = lda.components_[0]
second_topic = lda.components_[1]
third_topic = lda.components_[2]
fourth_topic = lda.components_[3]

firstcloud = WordCloud(
                      background_color='black',
                      width=2500,
                      height=1800
                     ).generate(" ".join(first_topic))
plt.imshow(firstcloud)
plt.axis('off')
plt.show()

Answer 1

你可以試試這個：

def create_wordcloud(model, topic):
    text = {word: value for word, value in model.show_topic(topic)}
    wc = WordCloud(background_color="white", max_words=1000)
    wc.generate_from_frequencies(text)
    plt.imshow(wc, interpolation="bilinear")
    plt.axis("off")
    plt.title("Topic" + " "+ str(topic))
    plt.show()

然后調用這個函數如下：

for i in range(1,num_topics):
    create_wordcloud(lda_model, topic=i)