spsanderson / healthyR.ai

healthyR.ai - AI package for the healthyverse
http://www.spsanderson.com/healthyR.ai/
Other
16 stars 6 forks source link

Enhancement `hai_default_classification_metric_set()` #222

Closed spsanderson closed 2 years ago

spsanderson commented 2 years ago

Function:

hai_default_classification_metric_set <- function(){

  yardstick::metric_set(
    yardstick::sensitivity,
    yardstick::specificity,
    yardstick::recall,
    yardstick::precision,
    yardstick::mcc,
    yardstick::accuracy,
    yardstick::f_meas,
    yardstick::kap,
    yardstick::ppv,
    yardstick::npv,
    yardstick::bal_accuracy
  )
}

Example:

> hai_default_classification_metric_set()
# A tibble: 11 x 3
   metric       class        direction
   <chr>        <chr>        <chr>    
 1 sensitivity  class_metric maximize 
 2 specificity  class_metric maximize 
 3 recall       class_metric maximize 
 4 precision    class_metric maximize 
 5 mcc          class_metric maximize 
 6 accuracy     class_metric maximize 
 7 f_meas       class_metric maximize 
 8 kap          class_metric maximize 
 9 ppv          class_metric maximize 
10 npv          class_metric maximize 
11 bal_accuracy class_metric maximize