根据模型的数学原理进行简单的代码自我复现以及使用测试,仅作自我学习用。模型原理此处不作过多赘述,仅罗列自己将要使用到的部分公式。
如文中或代码有错误或是不足之处,还望能不吝指正。
本文的思路借鉴了GMM算法的实现_MagicGeek的博客-优快云博客_gmm算法实现
有时某些数据不是简单地服从单个正态分布,而是服从由多个正态分布线性组合的复合分布。此时我们便使用GMM模型分析其内部的正态分布模型,将数据分为多个服从正态分布的类。
其中,为第k个正态分布的权重系数。
我们使用EM算法估算各个正态分布的期望方差矩阵
。E步通过
和
估算后验概率
,M步通过
的最大化对数似然函数来倒推出新的
与
。
对数似然函数为:
对其和
分别求偏导,得

本文介绍了GMM(高斯混合模型)的原理和实现,包括初始化、E步和M步的步骤,并通过实例展示了如何使用自制数据集进行训练和预测。结果显示了模型对复杂数据分布的有效拟合能力。
最低0.47元/天 解锁文章
1072

被折叠的 条评论
为什么被折叠?



