ComeçarComece de graça

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

Ver curso

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)
Editar e executar o código