predict.kmeans <- function(km, data) {k <- nrow(km$centers) n <- nrow(data) d <- as.matrix(dist(rbind(km$centers, data)))[-(1:k),1:k] out <- apply(d, 1, which.min) return(out)}