Customize your model assessment
Creating custom assessment functions is quite convenient when iterating through various models. The metric_set() function from the yardstickpackage can help you to achieve this.
Define a function that returns roc_auc, accuracy, sens(sensitivity) and specificity spec(specificity) and use it to assess your model.
The augmented data frame lr_augis already loaded and ready to go.
This exercise is part of the course
Feature Engineering in R
Exercise instructions
- Define a custom assessment function that returns
roc_auc,accuracy,sens, andspec. - Assess your model using your new function on
lr_augto obtain the metrics you just chose.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Define a custom assessment function
class_evaluate <- ___(___, ___, sens, ___)
# Assess your model using your new function
___(___, truth = Attrition,
estimate = .pred_class,
.pred_No)