CommencerCommencer gratuitement

Transformation de Box-Cox

En utilisant le jeu de données attrition_num contenant uniquement des données numériques sur les employés qui ont quitté l’entreprise, vous souhaitez construire un modèle capable de prédire si un employé est susceptible de rester, à partir de Attrition, une variable binaire encodée comme un factor. Pour rapprocher les distributions des variables d’une loi normale, vous allez créer une recette qui applique la transformation Box-Cox.

Les données attrition_num, la régression logistique lr_model, la fonction utilisateur class-evaluate() et les partitions train et test sont déjà chargées pour vous.

Cet exercice fait partie du cours

Feature engineering en R

Afficher le cours

Instructions

  • Créez une recette qui applique Box-Cox à toutes les variables numériques, y compris la cible.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Create a recipe that uses Box-Cox to transform all numeric features
lr_recipe_BC <- 
  recipe(Attrition ~., data = train) %>%
  ___(___)

lr_workflow_BC <- workflow() %>%
  add_model(lr_model) %>%
  add_recipe(lr_recipe_BC)
lr_fit_BC <- lr_workflow_BC %>%
  fit(train)
lr_aug_BC <-
  lr_fit_BC %>% augment(test)
lr_aug_BC %>% class_evaluate(truth = Attrition,
                 estimate = .pred_class,.pred_No)
Modifier et exécuter le code