学习LDA模型背后的数学原理,参考的文章为《LDA数学八卦》,文章中讲到马氏链及其平稳分布,觉得很有意思,也容易理解,就用Java验证了一下作者的数据及结论。现把代码贴出来,供以后的学习参考。
public class Main { private static final double[][] tm={ {0.65, 0.28, 0.07}, {0.15, 0.67, 0.18}, {0.12, 0.36, 0.52}}; private static final double[] pi={0.21, 0.68, 0.11}; /* * 根据初始概率和转移矩阵计算n代后各阶层的分布情况 * */ public static double[] nGen(int n){ double[][] temp=new double[2][3]; temp[0]=pi.clone(); int cur=1;int last=0; while(--n>=0){ for(int j=0;j<3;j++){ temp[cur][j]=0.0f; for(int k=0;k<3;k++){ temp[cur][j]+=temp[last][k]*tm[k][j]; } } last=cur; cur=(cur+1)%2; } return temp[last]; } public static void main(String[] args) { double[] d=nGen(1); for (int i = 0; i < d.length; i++) { System.out.println(d[i]); } }}
其实没有太复杂的东西:线性代数的中矩阵乘法而已,再加上滚动数组的一点点编码技巧,仅此而已!