PAM Clustering - Use the results in another data-set

1.9k views Asked by At

I've successfully run a Partitioning Around Medoids using the pam function (cluster package in R) and now, I would like to use the results to attribute new observations to the previously defined clusters/medoids.

Another way to put the problem is, given the k clusters/medoids that have been found by the pam function, which is closer to an additional observation that was not in the initial dataset?

x<-matrix(c(1,1.2,0.9,2.3,2,1.8,
            3.2,4,3.1,3.9,3,4.4),6,2)
x
     [,1] [,2]
[1,]  1.0  3.2
[2,]  1.2  4.0
[3,]  0.9  3.1
[4,]  2.3  3.9
[5,]  2.0  3.0
[6,]  1.8  4.4
pam(x,2)

Observations 1, 3 and 5, and 2, 4 and 6 are clustered together and observations 1 and 6 are the medoids:

Medoids:
     ID        
[1,]  1 1.0 3.2
[2,]  6 1.8 4.4
Clustering vector:
[1] 1 2 1 2 1 2

Now, to which cluster/medoid y should be attributed/associated with?

y<-c(1.5,4.5)

Oh, and in case you have several solutions, computing time matters in the big data-set I have.

1

There are 1 answers

2
Sandipan Dey On BEST ANSWER

Try this for k clusters in general:

k <- 2 # pam with k clusters
res <- pam(x,k)

y <- c(1.5,4.5) # new point

# get the cluster centroid to which the new point is to be assigned to
# break ties by taking the first medoid in case there are multiple ones

# non-vectorized function
get.cluster1 <- function(res, y) which.min(sapply(1:k, function(i) sum((res$medoids[i,]-y)^2)))

# vectorized function, much faster
get.cluster2 <- function(res, y) which.min(colSums((t(res$medoids)-y)^2))

get.cluster1(res, y)
#[1] 2
get.cluster2(res, y)
#[1] 2

# comparing the two implementations (the vectorized function takes much les s time)
library(microbenchmark)
microbenchmark(get.cluster1(res, y), get.cluster2(res, y))

#Unit: microseconds
#                 expr    min     lq     mean median     uq     max neval cld
# get.cluster1(res, y) 31.219 32.075 34.89718 32.930 33.358 135.995   100   b
# get.cluster2(res, y) 17.107 17.962 19.12527 18.817 19.245  41.483   100  a 

Extension to any arbitrary distance function:

# distance function
euclidean.func <- function(x, y) sqrt(sum((x-y)^2))
manhattan.func <- function(x, y) sum(abs(x-y))

get.cluster3 <- function(res, y, dist.func=euclidean.func) which.min(sapply(1:k, function(i) dist.func(res$medoids[i,], y)))
get.cluster3(res, y) # use Euclidean as default
#[1] 2
get.cluster3(res, y, manhattan.func) # use Manhattan distance
#[1] 2