pyanno4rt.learning_model.evaluation.metrics._kpi

KPI computation.

Overview

Function

kpi(true_labels, predicted_labels, thresholds)

Compute the model KPIs.

Functions

pyanno4rt.learning_model.evaluation.metrics._kpi.kpi(true_labels, predicted_labels, thresholds=(0.5, 0.5))[source]

Compute the model KPIs.

Parameters:

  • true_labels (ndarray) – Ground truth label values.

  • predicted_labels (tuple) – Tuple of arrays with the training and out-of-folds labels predicted by the machine learning outcome model.

  • thresholds (tuple, default=(0.5, 0.5)) – Probability thresholds for the binarization of the probability predictions.

Returns:

scores – Dictionary with the training and out-of-folds KPIs.

Return type:

dict