Comece agoraComece grátis

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 exercicio faz parte do curso

Feature Engineering em R

Ver curso

Instruções do exercicio

  • Crie uma receita que use Yeo-Johnson para transformar todos os recursos numéricos, incluindo o alvo.

exercicio interativo prático

Tente este exercicio 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)
Editar e Executar Código