LDA模型用pyLDAvis进行可视化遇到的问题及解决方法

报错信息如下：

TypeError:drop() takes from 1 to 2 positional arguments but 3 were given

 

 出错代码行如下：

vis_data = pyLDAvis.gensim.prepare(lda_model, corpus, dictionary)

 

错误原因：

过高版本的pandas中的drop函数只允许两个参数，因此报错

 

解决方法：

安装1.3.3版本的pandas 和 3.1.1版本的pyLDAvis

pip install pandas==1.3.3
pip install pyLDAvis==3.1.1

最后导入相应的库，运行成功！

import pandas as pd
import pyLDAvis.gensim_models

vis_data = pyLDAvis.gensim_models.prepare(best_ldamodel, corpus, dictionary)