一种潜在dirichlet分配的实现
ktLDA的Python项目详细描述
ktlda
这是为教学目的而实现的潜在dirichlet分配。
依赖性
- 努比
- 全面质量管理
示例
fromktldaimportKtLDAimportpicklewithopen('ourdata-cleaned.pickle','rb')asf:comp,rec=pickle.load(f)X=comp+recY=[0]*len(comp)+[1]*len(rec)lda=KtLDA(n_components=2,alpha=0.5,beta=0.5,iterations=10,max_vocab=5000,random_state=663)lda.fit(X)print(lda.doc_topic_dist)