CommencezCommencez gratuitement

Transformation de Yeo-Johnson

En utilisant l'ensemble de données attrition_num qui contient uniquement des données numériques sur les employé·e·s ayant quitté l'entreprise, vous voulez construire un modèle capable de prédire si une personne est susceptible de rester, en utilisant Attrition, une variable binaire de type factor. Pour rendre les variables plus proches d'une distribution normale, vous allez créer une recette qui applique la transformation Yeo-Johnson.

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

Cette activité fait partie du cours

Ingénierie des caractéristiques en R

Voir le cours

Instructions de l’exercice

  • Créez une recette qui utilise Yeo-Johnson pour transformer toutes les variables numériques, y compris la cible.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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

lr_workflow_YJ <- workflow() %>%
  add_model(lr_model) %>%
  add_recipe(lr_recipe_YJ)
lr_fit_YJ <- lr_workflow_YJ %>%
  fit(train)
lr_aug_YJ <-
  lr_fit_YJ %>% augment(test)
lr_aug_YJ %>% class_evaluate(truth = Attrition,
                 estimate = .pred_class,.pred_No)
Modifier et exécuter le code