Feb, 2024
混合模型:一个用于基于模型的聚类的Python一站式库,使用各种混合模型
Mixture-Models: a one-stop Python Library for Model-based Clustering
using various Mixture Models
TL;DR《Mixture-Models》是一个开源的Python库,用于拟合高斯混合模型及其变种,如Sparimonious GMM、因子分析混合模型、MClust模型、学生t分布混合模型等。它使用各种一阶/二阶优化算法,如梯度下降和牛顿法,通过自动微分工具简化了这些模型的实现和分析,从而帮助扩展这些模型到高维数据。该库提供了用户友好的模型评估工具,如BIC、AIC和对数似然估计。该软件包具有很高的可扩展性,允许用户轻松地添加新的分布和优化技术。我们进行了大规模模拟实验,比较了各种基于梯度的方法与期望最大化算法在各种设置下的性能,并确定相应最适合的方法。