Transformación de Yeo-Johnson
Usando el conjunto de datos attrition_num con toda la información numérica sobre las personas que han dejado la empresa, quieres crear un modelo que pueda predecir si es probable que una persona se quede, usando Attrition, una variable binaria codificada como factor. Para que las variables se comporten de forma casi normal, crearás una receta que aplique la transformación Yeo-Johnson.
Los datos attrition_num, el modelo de regresión logística lr_model, la función definida por el usuario class-evaluate() y las particiones train y test ya están cargados para ti.
Este ejercicio forma parte del curso
Ingeniería de características en R
Instrucciones del ejercicio
- Crea una receta que use Yeo-Johnson para transformar todas las variables numéricas, incluido el objetivo.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# 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)