ComenzarEmpieza gratis

¿Cuál es el predictor principal?

Has obtenido una predicción excelente, pero ¿cuáles fueron los principales predictores? ¿Cómo puedes interpretar el modelo para ir más allá de los resultados en bruto? A los modelos de Machine Learning a menudo se les critica por su falta de interpretabilidad. Sin embargo, los rankings de importancia de variables arrojan luz sobre la relevancia de las características que has elegido respecto al resultado. Así que vamos a investigar la importancia de variables y partimos de ahí.

Este ejercicio forma parte del curso

Ingeniería de características en R

Ver curso

Instrucciones del ejercicio

  • Crea un gráfico de importancia de variables.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

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 y ejecutar código