R/predict_helpers.R

In kbodwin/celery: What the Package Does (One Line, Title Case)

stats_kmeans_predict <- function(object, new_data) {
  res <- apply(flexclust::dist2(object$centers, new_data), 2, which.min)
  factor(res, levels = seq_along(object$size))
}

clusterR_kmeans_predict <- function(object, new_data) {
  res <- apply(flexclust::dist2(object$centroids, new_data), 2, which.min)
  factor(res, levels = seq_along(object$obs_per_cluster))
}