如何设置Spark Kmeans的初始中心

我在使用Spark ML运行Kmeans。我有一组数据和三个现有的中心,例如这三个中心是:[1.0,1.0,1.0],[5.0,5.0,5.0],[9.0,9.0,9.0]. 那么,我如何指定Kmeans的中心为上述三个向量呢?我看到Kmeans对象有一个seed参数,但seed参数是long类型而不是数组。那么,我如何告诉Spark Kmeans只使用现有的中心进行聚类呢?

或者说,我不理解Spark Kmeans中的seed是什么意思,我认为seeds应该是一个表示指定中心的向量数组,在运行聚类之前使用。


回答:

实际上,seed并不是你所想的意思,即它不是用来“播种”(初始化)聚类中心的,而是简单地设置随机种子 – 你可以在ScalaPython API的文档中确认这一点。

据我所知,目前(Spark 2.1)在Spark ML中没有办法为k-means提供初始聚类中心(参见这个回答关于Spark MLlib)。根据文档,initMode参数:

可以是“random”以随机选择点作为初始聚类中心,或者是“k-means||”以使用k-means++的并行变体

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注