Get startedGet started for free

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

View Course

Exercise instructions

  • Define a custom assessment function that returns roc_auc, accuracy, sens, and spec.
  • 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)
Edit and Run Code