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