Receta básica
Usando el conjunto de datos attrition_num con todos los datos numéricos sobre empleados que han dejado la empresa, quieres construir un modelo que pueda predecir si es probable que un empleado se quede, usando Attrition, una variable binaria codificada como factor. Para empezar, vas a definir una receta básica que no hace nada más que establecer la fórmula del modelo y los datos de entrenamiento.
Los datos de attrition_num, la 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 básica definiendo solo la fórmula del modelo.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Create a plain recipe defining only the model formula
lr_recipe_plain <-
___(___ ~., ___ = ___)
lr_workflow_plain <- workflow() %>%
add_model(lr_model) %>%
add_recipe(lr_recipe_plain)
lr_fit_plain <- lr_workflow_plain %>%
fit(train)
lr_aug_plain <-
lr_fit_plain %>% augment(test)
lr_aug_plain %>% class_evaluate(truth = Attrition,
estimate = .pred_class,.pred_No)