集群仍然按与中心的距离来分配,除了 k-medoids,中心实际上是数据集中的一个数据点。请参阅下面的 R 代码:
library(ClusterR)
library(ggplot2)
set.seed(100)
# we use the iris data set, split into 2
a = sample(nrow(iris),90)
data_b = iris[-a,1:4]
data_a = iris[a,1:4]
#perform k medoids
cm = Cluster_Medoids(data_a,clusters=3)
您可以看到中心点是数据点:
cm$medoids
Sepal.Length Sepal.Width Petal.Length Petal.Width
95 5.6 2.7 4.2 1.3
12 4.8 3.4 1.6 0.2
111 6.5 3.2 5.1 2.0
我们继续预测:
pm = predict_Medoids(data_b,MEDOIDS=cm$medoids)
我们可以从第一个数据集计算中心点之间的距离,并将第二个数据集分配给集群:
M = as.matrix(dist(rbind(cm$medoids,data_b)))
labs = sapply(4:nrow(M),function(i)which.min(M[i,1:3]))
我们检查,您可以看到,手动计算的集群与 clusterR 中实现的一致:
table(pm$clusters==labs)
TRUE
60
我们可以想象一下:
PCA = prcomp(rbind(data_a,data_b))$x
plotdf = data.frame(PCA[,1:2],
label=c(cm$clusters,pm$clusters),
dataset=rep(c("train","pred"),c(nrow(data_a),nrow(data_b)))
)
ggplot(plotdf,aes(x=PC1,y=PC2,col=factor(label),shape=dataset)) +
geom_point() + scale_color_brewer(palette="Paired") + theme_bw()