首页 > 解决方案 > 带有`agnes`的集群:如何获得集群成员资格

问题描述

我有一个巨大的数据集并创建了agnes如下集群:

hc.res1 <- agnes(example, method = "complete")
plot(hc.res1, which.plots = 2, main = "Complete Linkage (agnes)")

在此处输入图像描述

我想将数据集分成两个集群(一个拆分)。我该怎么做agnes?所以我想得到一个如下所示的列表

| Datarow | Cluster |
| 1       | 1       |
| 2       | 2       |
| 3       | 1       |

标签: rhierarchical-clustering

解决方案


您可以使用该cutree函数将树切割成若干 k 个簇。iris由于没有给出示例数据,让我们使用包含 3 个组的内置数据集来演示这一点:

library(cluster)

example <- iris # sample data set from R
hc.res1 <- agnes(example, method = "complete")
plot(hc.res1, which.plots = 2, main = "Complete Linkage (agnes)")

# let's assume k=3 clusters, then cutree shows the groups
cutree(k=3, hc.res1)

返回的数字cutree对应于原始数据集的行号。


推荐阅读