kmeans

语法

kmeans(X, k, [maxIter=300], [randomSeed], [init='random'])

参数

X 是一个表,表示训练集。

k 是一个正整数,表示要生成的聚类数。

maxIter 是一个正整数,表示质心更新的最大迭代次数。默认值是300。

randomSeed 是一个整数,表示质心初始化时随机算法的种子。默认值为 NULL。

init 可以是一个字符串或者一个矩阵,表示初始值的选择方式。默认值是 'random'。

  • init 是一个字符串,则可选参数为 'random' 或 'k-means++'。'random' 表示根据 randomSeed 随机生成,'kmeans++' 表示根据 kmeans++ 算法生成。

  • init 是一个矩阵,表示自定义的质心。其列数与表 X 需保持一致,行数为 k

详情

对训练集执行 K-Means 聚类。返回的结果是一个字典,包含以下 key:

  • centers:一个 k 行 m 列的矩阵(m 是 X 的列数),包含各个类的质心坐标。

  • predict:一个聚类模型预测函数,数据类型是 FUNCTIONDEF。

  • modelName:字符串 'KMeans'。

  • model:保存的模型,数据类型为 RESOURCE,用于预测。

  • labels:一个向量,表示 X 中每一行数据对应的聚类的类标签。

例子

使用模拟数据训练一个 K-Means 模型:

$ t = table(100:0, `x0`x1, [DOUBLE, DOUBLE])
$ x0 = norm(1.0, 1.0, 50)
$ x1 = norm(1.0, 1.5, 50)
$ insert into t values (x0, x1)
$ x0 = norm(2.0, 1.0, 50)
$ x1 = norm(-1.0, 1.5, 50)
$ insert into t values (x0, x1)
$ x0 = norm(-1.0, 1.0, 50)
$ x1 = norm(-3.0, 1.5, 50)
$ insert into t values (x0, x1);

$ model = kmeans(t, 3);
$ model;

centers->

#0        #1
--------- ---------
-1.048027 -3.809539
1.110899  1.24216
1.677974  -1.19158

predict->kmeansPredict
modelName->KMeans
model->KMeans
labels->[2,2,2,2,2,2,3,2,3,2,...]