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
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)