IniziaInizia gratis

Qual è il principale predittore?

Hai ottenuto un'ottima previsione, ma quali sono stati i predittori principali? Come puoi dare un senso al modello per andare oltre i risultati grezzi? I modelli di machine learning sono spesso criticati per la scarsa interpretabilità. Tuttavia, le classifiche di importanza delle variabili fanno luce sulla rilevanza delle feature scelte rispetto all'outcome. Indaghiamo quindi l'importanza delle variabili e partiamo da lì.

Questo esercizio fa parte del corso

Feature Engineering in R

Visualizza il corso

Istruzioni dell'esercizio

  • Crea un grafico di importanza delle variabili.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

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)
Modifica ed esegui il codice