使用Mallet扩展LDA模型

Question

我试图通过添加另一层位置来扩展LDA模型。 是否可以向Mallet添加另一个图层？如果是这样，我应该扩展哪些类？

我尝试建模的过程： 1.选择一个地区 2.选择一个主题 3.选择一个单词

Answer 1

cc.mallet.topics.SimpleLDA类旨在用作开发新模型的基础：https://github.com/mimno/Mallet/blob/master/src/cc/mallet/topics/SimpleLDA.java

可能有从头开始设计新模型的替代方案。如果region完全确定主题分布，并且每个文档来自一个区域，则可以简单地合并区域中的所有文档。如果每个文档都有一个或多个区域，则可以将区域视为“作者”并实现Author-Topic模型。如果您想测量区域和主题之间更间接的关系，您可以尝试使用Dirichlet-Multinomial Regression（DMR）模型。