step_poly()
Maintenant que vous avez une référence, vous pouvez comparer les performances de votre modèle en ajoutant une transformation polynomiale à toutes les valeurs numériques.
Les données attrition_num, le modèle de régression logistique lr_model, la fonction utilisateur class-evaluate(), ainsi que les partitions train et test ont déjà été chargés pour vous.
Cet exercice fait partie du cours
Feature engineering en R
Instructions
- Ajoutez une transformation polynomiale à tous les prédicteurs numériques.
- Ajustez le workflow sur les données d’entraînement.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
lr_recipe_poly <-
recipe(Attrition ~., data = train) %>%
# Add a polynomial transformation to all numeric predictors
___
lr_workflow_poly <- workflow() %>%
add_model(lr_model) %>%
add_recipe(lr_recipe_poly)
# Fit workflow to the train data
lr_fit_poly <- ___ %>% fit(train)
lr_aug_poly <- lr_fit_poly %>% augment(test)
lr_aug_poly %>% class_evaluate(truth = Attrition, estimate = .pred_class,.pred_No)