Skip to content

Extract clusters from model

Usage

extract_centroids(object, ...)

Arguments

object

An cluster_spec object.

...

Other arguments passed to methods.

Value

A tibble::tibble() with 1 row for each centroid and their position.

Examples

set.seed(1234)
kmeans_spec <- k_means(num_clusters = 5) %>%
  set_engine("stats")

kmeans_fit <- fit(kmeans_spec, ~., mtcars)

kmeans_fit %>%
  extract_centroids()
#> # A tibble: 5 × 12
#>   .cluster    mpg   cyl  disp    hp  drat    wt  qsec    vs    am  gear
#>   <fct>     <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
#> 1 Cluster_1  19.9  5.71  167. 120.   3.71  3.11  18.5 0.571 0.429   4  
#> 2 Cluster_2  14.6  8     340. 272.   3.68  3.54  15.1 0     0.5     4  
#> 3 Cluster_3  13.7  8     443  206.   3.06  4.97  17.6 0     0       3  
#> 4 Cluster_4  27.0  4     102.  81.4  4.09  2.20  18.8 0.9   0.8     4.1
#> 5 Cluster_5  17.1  7.71  295. 161.   3.05  3.60  17.7 0.143 0       3  
#> # … with 1 more variable: carb <dbl>