diff --git a/R/define_target.R b/R/define_target.R index 2d8bff9..dbb9fd7 100644 --- a/R/define_target.R +++ b/R/define_target.R @@ -72,7 +72,7 @@ prepare_data_for_modelling <- function(data, target){ mydata <- mydata %>% recipes::recipe( ~ .) %>% - recipes::step_dummy(recipes::all_predictors(), -recipes::all_numeric(), -target$id_variable, -target$target_variable, one_hot = T) %>% + recipes::step_dummy(recipes::all_nominal_predictors(), -target$id_variable, -target$target_variable, one_hot = T) %>% recipes::prep() %>% recipes::bake(mydata) %>% mutate_if(is.logical, as.integer)