文章目录
- 作者主题模型(ATM)
- python代码示例
作者主题模型(ATM)
Gensim
中的主题模型包括三种,分别是LDA (Latent Dirichlet Allocation)
主题模型、加入了作者因素的作者主题模型 (Author-Topic Model, ATM)
和加入了时间因素的动态主题模型 (Dynamic Topic Models, DTM)
。
作者主题模型(ATM)
的输入除了分词后的文章内容,还包括作者和文章的对应关系;模型的输出为每位作者对于每个主题(主题数n
可以自己设定)的倾向度。
python代码示例
- 导入相关库
from gensim.corpora import Dictionary
from gensim.models