Transformação de Yeo-Johnson
Usando o conjunto de dados attrition_num com todos os dados numéricos sobre funcionários que deixaram a empresa, você quer criar um modelo que preveja se um funcionário provavelmente vai permanecer, usando Attrition, uma variável binária codificada como factor. Para que os recursos se comportem quase como normais, você vai criar uma receita que implemente a transformação Yeo-Johnson.
Os dados attrition_num, o modelo de regressão logística lr_model, a função definida pelo usuário class-evaluate() e as divisões train e test já foram carregados para você.
Este exercício faz parte do curso
Feature Engineering em R
Instruções do exercício
- Crie uma receita que use Yeo-Johnson para transformar todos os recursos numéricos, incluindo o alvo.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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)