Slutföra modellen
Nu är det dags att omsätta resultaten från ditt justeringsarbete i praktiken och imponera på HR-teamet. Du kan slutföra din modell med det optimala straffvärdet och se om den lever upp till förväntningarna. Dina resultat är inlästa och den egendefinerade funktionen class_evaluate() finns tillgänglig i din miljö.
Den här övningen är en del av kursen
Feature Engineering i R
Övningsinstruktioner
- Välj det optimala straffvärdet för Lasso.
- Träna en slutlig modell med det optimala straffvärdet.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# Select the optimal penalty for the Lasso
best_penalty <- ___(tune_output, metric = 'roc_auc', desc(penalty))
best_penalty
# Fit a final model using the optimal penalty
final_fit <- ___(workflow_lasso_tuned, best_penalty) %>%
fit(data = train)
final_fit %>% tidy()
final_fit %>% augment(test) %>% class_evaluate(truth = Attrition,
estimate = .pred_class,
.pred_Yes)