search

uclamii / model_tuner

A library to tune the hyperparameters of common ML models. Supports calibration and custom pipelines.
Apache License 2.0
3 stars 0 forks source link

Report Model Metrics #54

Closed lshpaner closed 1 week ago

lshpaner commented 3 weeks ago

Let us print the following additional model output alongside existing classification_report(), confusion_matrix() :

  Precision/PPV
  Average Precision
  Sensitivity
  Specificity
  AUC ROC
  Brier Score

To this end, we can create a standalone function to be used inside a standalone script to work w/ the library and call the function report_model_metrics()