ComenzarEmpieza gratis

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

Ver curso

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)
Editar y ejecutar código