CommencerCommencer gratuitement

Quel est le principal prédicteur ?

Votre prédiction est remarquable, mais quels étaient les principaux prédicteurs ? Comment interpréter le modèle pour aller au-delà des résultats bruts ? Les modèles de Machine Learning sont souvent critiqués pour leur manque d’interprétabilité. Toutefois, le classement de l’importance des variables éclaire la pertinence des caractéristiques que vous avez choisies par rapport à la variable cible. Examinons donc l’importance des variables et avançons à partir de là.

Cet exercice fait partie du cours

Feature engineering en R

Afficher le cours

Instructions

  • Créez un graphique d’importance des variables.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

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)
Modifier et exécuter le code