don't drop levels in hier_clust prediction

EmilHvitfeldt · EmilHvitfeldt · commit f6ff8224ff5c · 2023-08-31T11:03:07.000-07:00
diff --git a/R/predict_helpers.R b/R/predict_helpers.R
@@ -150,5 +150,5 @@ make_predictions <- function(x, prefix, n_clusters) {
   }
   pred_clusts <- unique(clusters$.cluster)[pred_clusts_num]
 
-  return(factor(pred_clusts))
+  pred_clusts
 }

Original file line number	Diff line number	Diff line change
`@@ -150,5 +150,5 @@ make_predictions <- function(x, prefix, n_clusters) {`
`150`	`150`	`}`
`151`	`151`	`pred_clusts <- unique(clusters$.cluster)[pred_clusts_num]`
`152`	`152`
`153`		`- return(factor(pred_clusts))`
	`153`	`+ pred_clusts`
`154`	`154`	`}`