testAUC 0.0.3

testAUC is a library of tools to evaluate the true performance of ML/AI models

testAUC

Quick start

pip install testAUC

To evaluate the drift between the Validation set to the Test set use:

from testAUC import roc_drift
tpr_drift, fpr_drift, drift, thresholds_val, mean_drift = roc_drift(y_true_val, y_score_val, y_true_tst, y_score_tst)

Motivation

The ROC (Receiver operating characteristic) curve is generally used for evaluation of ML/AI model performance in classification tasks. There are a number of pitfalls of using the ROC curve, some of which are outlined in example below

Example 0: example0.py

Demonstrates how two very different predictions can result in the exact same ROC curve.

Example 1: example1.py

Demonstrates the shifting of the operation point, despite the favourable AUC. We show predictions of a (simulated) model on the Validation set and the Test set. The ROC AUC "happens" to be identical. However, if you choose a threshold on the Validation set, aiming for a Sensitivity/Specificity operation point, you would get completely different operation point on the Test set

Example 2: example2.py

Demonstrates a measure of robustness to noise. We show two models, Initially it seems that Model 1 is far better, offering ROC AUC above 91% However, evaluating the robustness to uniform noise shows Model 2 is more robust

Example 3: example3.py

Demonstrates a measure of DRIFT between the validation set and the test set We show two models, both of them "happen" to have the exact same AUC on the validation and test sets. However, evaluating the DRIFT of the Sensitivity/Specificity operation point reveals that Model 1 is better (although in this extreme case, both models exhibit significant drifts)

Note that at some operation point, Model 2 actually exceeds a drift of -50% from the Sensitivity expected at Validation time, compared to the Sensitivity received at test time - without affecting the total AUC.