简体   繁体   中英

R k-means algorithm custom centers

I have a 2D dataset imported in R - (x,y) coordinates. I want to perform k-means clustering on this dataset, but I would like to set concrete coordinates to be initial centers. For example, I would like to start with 5 centers with their values to be (5, 10), (3, 8), (46, 22), (87, 66), (39, 41) .

I saw a centers parameter in kmeans function, but I do not understand how to set my values to be centeres.

kmeans(data, centers = ...) # what to set here?

The centers parameter takes either an integer k , in which case k random points from data are chosen as initial centers, or a matrix of initial centers, with as many columns as data . Try this:

x <- rbind(matrix(rnorm(100, sd = 0.3), ncol = 2),
                 matrix(rnorm(100, mean = 1, sd = 0.3), ncol = 2))
kmeans(x,centers=3)
kmeans(x,centers=x[1:3,])

Just pass a matrix , here is a quick example:

data = matrix(c(1.1,1,0.97,0.99,0.95,0.8,0.91,2.1,2,2.4,4.1,4.4,4.5,3.9,1.5,1.2,1.7,2.6,2.7,2.44), ncol=2)

Now let's mention 2 starting points C1 (x1=1,y1=3) and C2 (x2=2,y2=4) (even if there are obviously 3 groups):

km = kmeans(data, centers=matrix(c(1,2,3,4),ncol=2))

Some plotting after applying the algo: df = transform(as.data.frame(data), group=as.character(km$cluster)) library(ggplot2)

ggplot(df, aes(V1, V2, color=group)) + geom_point()

在此输入图像描述

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM