如何加快Scikit学习的k均值？

Question

在我的项目中，我曾使用k均值对组之间的数据进行分类，但是我对Scikit-learn的k均值的计算存在问题-速度非常慢。我需要加强它。

我尝试将更n_jobs改为-1，但是速度仍然很慢！

有什么建议如何加快？

Answer 1

scikit-learn的主要解决方案是切换到微型批处理kmeans，这大大减少了计算资源。在某种程度上，它是SGD（随机梯度下降）与GD（梯度下降）的相似方法，用于优化非线性函数-SGD通常更快（就收敛到局部解所需的计算周期而言）。请注意，这给优化带来了更多差异，因此结果可能更难以重现（优化将以“全批” kmeans的形式出现在不同的解决方案中）。