roc_auc

improvelib.metrics.roc_auc(y_true, y_prob)

Compute the Receiver Operating Characteristic AUC. Uses sklearn.metrics.roc_auc_score().

Used in compute_metrics and can be used directly.

Parameters:

y_truenp.ndarray

Ground truth.

y_probnp.ndarray

Target scores made by the model.

Returns:

roc_aucfloat

The computed Receiver Operating Characteristic AUC.

Example

This function is called by compute_metrics which is called by compute_performance_scores, which also saves the scores.

You can use this function directly, for example during model training like so:

import improvelib.metrics as metr
roc_auc_to_monitor = metr.roc_auc(y_true, y_pred)