ComeçarComece de graça

Qual é o principal preditor?

Você obteve uma previsão excelente, mas quais foram os principais preditores? Como fazer sentido do modelo para ir além dos resultados brutos? Modelos de Machine Learning costumam ser criticados pela falta de interpretabilidade. No entanto, os rankings de importância de variáveis ajudam a esclarecer a relevância das suas features em relação ao desfecho. Vamos investigar a importância das variáveis e avançar a partir daí.

Este exercício faz parte do curso

Feature Engineering em R

Ver curso

Instruções do exercício

  • Crie um gráfico de importância de variáveis.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

lr_fit <- lr_workflow %>%
  fit(test)

lr_aug <- lr_fit %>%
  augment(test)

lr_aug %>% class_evaluate(truth = Attrition,
                          estimate = .pred_class,
                          .pred_No)

# Create a variable importance chart
lr_fit %>%
  extract_fit_parsnip() %>%
  ___(aesthetics = list(fill = "steelblue"), num_features = 5)
Editar e executar o código