Ajuster et évaluer le modèle
Maintenant que vous avez traité les valeurs manquantes et créé des variables fictives, il est temps d’évaluer les performances de votre modèle !
Le jeu de données attrition, ainsi que les partitions test et train, la lr_recipe et votre logistic_model() déclaré sont déjà chargés pour vous.
Cet exercice fait partie du cours
Feature engineering en R
Instructions
- Regroupez le modèle et la recette dans un workflow.
- Ajustez le workflow sur les données d’entraînement.
- Générez un data frame augmenté pour évaluer les performances.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Bundle model and recipe in workflow
lr_workflow <- ___() %>%
add_model(lr_model) %>%
add_recipe(lr_recipe)
# Fit workflow to the train data
lr_fit <- ___(lr_workflow, data = train)
# Generate an augmented data frame for performance assessment
lr_aug <- lr_fit %>% ___(test)
lr_aug %>% roc_curve(truth = Attrition, .pred_No) %>% autoplot()
bind_rows(lr_aug %>% roc_auc(truth = Attrition, .pred_No),
lr_aug %>% accuracy(truth = Attrition, .pred_class))