CommencerCommencer gratuitement

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

Afficher le cours

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)
Modifier et exécuter le code