Feb, 2024

混合模型:一个用于基于模型的聚类的 Python 一站式库,使用各种混合模型

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