学习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]);
}
}
}
其实没有太复杂的东西:线性代数的中矩阵乘法而已,再加上滚动数组的一点点编码技巧,仅此而已!