minor tweak to within_cluster_sse

tidymodels · Jul 26, 2022 · 5a22a4e · 5a22a4e
1 parent ca58700
commit 5a22a4e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/R/metric-sse.R b/R/metric-sse.R
@@ -50,7 +50,8 @@ within_cluster_sse <- function(object, new_data = NULL,
         .cluster = factor(paste0("Cluster_", .cluster))
       ) %>%
       dplyr::group_by(.cluster) %>%
-      dplyr::summarize(wss = sum(dist))
+      dplyr::summarize(wss = sum(dist),
+                       n_obs = n())
   }
 
   return(res)