Transformação de Box-Cox
Usando o conjunto de dados attrition_num com todos os dados numéricos sobre funcionários que deixaram a empresa, você quer construir um modelo que possa prever se um funcionário tende a ficar, usando Attrition, uma variável binária codificada como factor. Para que as variáveis fiquem aproximadamente normais, você vai criar uma receita que implemente a transformação Box-Cox.
Os dados attrition_num, o modelo de regressão logística lr_model, a função definida pelo usuário class-evaluate() e as partições train e test já estão carregados para você.
Este exercício faz parte do curso
Feature Engineering em R
Instruções do exercício
- Crie uma receita que use Box-Cox para transformar todas as variáveis numéricas, incluindo a variável alvo.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Create a recipe that uses Box-Cox to transform all numeric features
lr_recipe_BC <-
recipe(Attrition ~., data = train) %>%
___(___)
lr_workflow_BC <- workflow() %>%
add_model(lr_model) %>%
add_recipe(lr_recipe_BC)
lr_fit_BC <- lr_workflow_BC %>%
fit(train)
lr_aug_BC <-
lr_fit_BC %>% augment(test)
lr_aug_BC %>% class_evaluate(truth = Attrition,
estimate = .pred_class,.pred_No)