step_poly()
Maintenant que vous avez une base de référence, vous pouvez comparer la performance de votre modèle en ajoutant une transformation polynomiale à toutes les valeurs numériques.
Les données attrition_num, la régression logistique lr_model, la fonction personnalisée class-evaluate() et les partitions train et test ont déjà été chargées pour vous.
Cette activité fait partie du cours
Ingénierie des caractéristiques en R
Instructions de l’exercice
- 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 ce code d’exemple.
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)