Trasformazione Yeo-Johnson
Usando il dataset attrition_num con tutti i dati numerici sui dipendenti che hanno lasciato l'azienda, vuoi costruire un modello in grado di prevedere se un dipendente probabilmente resterà, usando Attrition, una variabile binaria codificata come factor. Per far sì che le variabili si comportino in modo quasi normale, creerai una recipe che applica la trasformazione Yeo-Johnson.
I dati attrition_num, la regressione logistica lr_model, la funzione definita dall'utente class-evaluate() e gli split train e test sono già caricati per te.
Questo esercizio fa parte del corso
Feature Engineering in R
Istruzioni dell'esercizio
- Crea una recipe che usi Yeo-Johnson per trasformare tutte le variabili numeriche, incluso il target.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# 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)