Aan de slagGa gratis aan de slag

Tune random forest models

Now that you have a working logistic regression model you will prepare a random forest model to compare it with.

Deze oefening maakt deel uit van de cursus

Machine Learning in the Tidyverse

Cursus bekijken

Oefeninstructies

  • Use crossing() to expand the cross-validation data for values of mtry using the values of 2, 4, 8, and 16.
  • Build random forest models for each fold/mtry combination.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

library(ranger)

# Prepare for tuning your cross validation folds by varying mtry
cv_tune <- cv_data %>%
  crossing(mtry = c(___)) 

# Build a cross validation model for each fold & mtry combination
cv_models_rf <- cv_tune %>% 
  mutate(model = map2(___, ___, ~ranger(formula = Attrition~., 
                                           data = .x, mtry = .y,
                                           num.trees = 100, seed = 42)))
Code bewerken en uitvoeren