Ajustando e avaliando o modelo
Agora que você tratou os valores ausentes e criou variáveis dummy, é hora de avaliar o desempenho do seu modelo!
O conjunto de dados attrition, juntamente com as divisões test e train, a lr_recipe e o seu logistic_model() já estão carregados para você.
Este exercício faz parte do curso
Feature Engineering em R
Instruções do exercício
- Agrupe o modelo e a receita em um workflow.
- Ajuste o workflow aos dados de treino.
- Gere um data frame aumentado para avaliar o desempenho.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Bundle model and recipe in workflow
lr_workflow <- ___() %>%
add_model(lr_model) %>%
add_recipe(lr_recipe)
# Fit workflow to the train data
lr_fit <- ___(lr_workflow, data = train)
# Generate an augmented data frame for performance assessment
lr_aug <- lr_fit %>% ___(test)
lr_aug %>% roc_curve(truth = Attrition, .pred_No) %>% autoplot()
bind_rows(lr_aug %>% roc_auc(truth = Attrition, .pred_No),
lr_aug %>% accuracy(truth = Attrition, .pred_class))