准备工作
首先,确认你已经安装好pyldavis和gensim库。
pip install gensim
pip install pyldavis
代码修改
其次,如果报错not attribute 是因为pyldavis的版本不同,新版本需要做如下更改:把所有的pyLDAvis.gensim改成gensimvis即可。
import pyLDAvis.gensim as gensimvis
pyLDAvis.enable_notebook()
vis_sub_10 = gensimvis.prepare(lda_fst, corpus, dic, sort_topics = False)
vis_sub_20 = gensimvis.prepare(lda_snd, corpus, dic, sort_topics = False)
pyLDAvis.display(vis_sub_10)
成功!效果展示如下图:
或者你也可以安装低版本的pyldavis
pip install pyLDAvis==2.1.2
附原代码及参考链接
import pyLDAvis.gensim
pyLDAv