CommencezCommencez gratuitement

Créer un diagramme en boîtes à moustaches

caret offre diverses méthodes pour comparer des modèles. Toutes reposent sur la fonction resamples(). Ma préférée est le diagramme en boîtes à moustaches, qui permet de comparer la distribution de la justesse prédictive (ici l'AUC) pour les deux modèles.

En général, vous cherchez le modèle dont la médiane de l'AUC est plus élevée, ainsi qu'un intervalle min–max plus restreint.

Vous pouvez produire ce graphique avec la fonction bwplot(), qui trace un diagramme en boîtes et moustaches des scores hors échantillon du modèle. Les diagrammes en boîtes et moustaches affichent la médiane de chaque distribution sous forme de ligne, et l'intervalle interquartile sous forme de boîte autour de cette ligne médiane. Vous pouvez transmettre l'argument metric = "ROC" à la fonction bwplot() pour afficher les scores ROC hors échantillon du modèle et choisir celui avec la médiane ROC la plus élevée.

Si vous ne précisez pas la mesure à tracer, bwplot() en affichera automatiquement trois.

Cette activité fait partie du cours

Machine Learning avec caret en R

Voir le cours

Instructions de l’exercice

Passez l'objet resamples à la fonction bwplot() pour créer un diagramme en boîtes à moustaches. Examinez le graphique obtenu et notez quel modèle présente la médiane ROC la plus élevée. Assurez-vous d'indiquer la mesure que vous souhaitez tracer.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Create bwplot
Modifier et exécuter le code