Personnalisez l’évaluation de votre modèle
Créer des fonctions d’évaluation personnalisées est très pratique lorsque vous itérez sur différents modèles. La fonction metric_set() du package yardstick peut vous y aider.
Définissez une fonction qui renvoie roc_auc, accuracy, sens (sensibilité) et spec (spécificité), puis utilisez-la pour évaluer votre modèle.
Le data frame augmenté lr_aug est déjà chargé et prêt à l’emploi.
Cet exercice fait partie du cours
Feature engineering en R
Instructions
- Définissez une fonction d’évaluation personnalisée qui renvoie
roc_auc,accuracy,sensetspec. - Évaluez votre modèle avec votre nouvelle fonction sur
lr_augpour obtenir les métriques que vous venez de choisir.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Define a custom assessment function
class_evaluate <- ___(___, ___, sens, ___)
# Assess your model using your new function
___(___, truth = Attrition,
estimate = .pred_class,
.pred_No)