Skip to content

performance

Utility module for computing various performance metrics for machine learning models.

Classes:

Performance 

Performance(
    labels: Any, predictions: Any, metric: str = "rocauc"
)

Returns the value of a given metric.

Parameters

labels (np.array) : labels predictions (np.array) : predictions metric (str) : the metric to compute

Returns:

value (float) : the value of the metric

TODO we can add more metrics here

TODO currently for classification metrics like precision, recall, f1score and mcc, we are using a threshold of 0.5 to convert the probabilities to binary predictions. However for models with imbalanced predictions, where the meaningful threshold is not located at 0.5, one can end up with full of 0s or 1s, and thus meaningless performance metrics.

Parameters:

  • labels (Any) –

    Ground truth labels

  • predictions (Any) –

    Model predictions

  • metric (str, default: 'rocauc' ) –

    Type of metric to compute (default: "rocauc")

Methods:

  • data2array –

    Convert input data to numpy array.

  • f1score –

    Compute F1 score.

  • handle_multiclass –

    Handle the case of multiclass classification.

  • mcc –

    Compute Matthews Correlation Coefficient.

  • prauc –

    Compute PR AUC score.

  • precision –

    Compute precision score.

  • recall –

    Compute recall score.

  • rocauc –

    Compute ROC AUC score.

  • spearmanr –

    Compute Spearman correlation coefficient.

Source code in src/stimulus/utils/performance.py 
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
def __init__(self, labels: Any, predictions: Any, metric: str = "rocauc") -> None:
    """Initialize Performance class with labels, predictions and metric type.

    Args:
        labels: Ground truth labels
        predictions: Model predictions
        metric: Type of metric to compute (default: "rocauc")
    """
    labels_arr = self.data2array(labels)
    predictions_arr = self.data2array(predictions)
    labels_arr, predictions_arr = self.handle_multiclass(labels_arr, predictions_arr)
    if labels_arr.shape != predictions_arr.shape:
        raise ValueError(
            f"The labels have shape {labels_arr.shape} whereas predictions have shape {predictions_arr.shape}.",
        )
    function = getattr(self, metric)
    self.val = function(labels_arr, predictions_arr)

data2array 

data2array(data: Any) -> NDArray[float64]

Convert input data to numpy array.

Parameters:

  • data (Any) –

    Input data in various formats

Returns:

  • NDArray[float64] –

    NDArray[np.float64]: Converted numpy array

Raises:

  • ValueError –

    If input data type is not supported

Source code in src/stimulus/utils/performance.py 
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
def data2array(self, data: Any) -> NDArray[np.float64]:
    """Convert input data to numpy array.

    Args:
        data: Input data in various formats

    Returns:
        NDArray[np.float64]: Converted numpy array

    Raises:
        ValueError: If input data type is not supported
    """
    if isinstance(data, list):
        return np.array(data, dtype=np.float64)
    if isinstance(data, np.ndarray):
        return data.astype(np.float64)
    if isinstance(data, torch.Tensor):
        return data.detach().cpu().numpy().astype(np.float64)
    if isinstance(data, (int, float)):
        return np.array([data], dtype=np.float64)
    raise ValueError(f"The data must be a list, np.array, torch.Tensor, int or float. Instead it is {type(data)}")

f1score 

f1score(
    labels: NDArray[float64], predictions: NDArray[float64]
) -> float

Compute F1 score.

Source code in src/stimulus/utils/performance.py 
138
139
140
141
def f1score(self, labels: NDArray[np.float64], predictions: NDArray[np.float64]) -> float:
    """Compute F1 score."""
    predictions_binary = np.array([1 if p > BINARY_THRESHOLD else 0 for p in predictions])
    return float(f1_score(labels, predictions_binary))

handle_multiclass 

handle_multiclass(
    labels: NDArray[float64], predictions: NDArray[float64]
) -> tuple[NDArray[float64], NDArray[float64]]

Handle the case of multiclass classification.

Parameters:

  • labels (NDArray[float64]) –

    Labels array of shape (N,) or (N, 1)

  • predictions (NDArray[float64]) –

    Predictions array of shape (N,) or (N, C) where C is number of classes

Returns:

  • tuple[NDArray[float64], NDArray[float64]] –

    tuple[NDArray[np.float64], NDArray[np.float64]]: Processed labels and predictions

Raises:

  • ValueError –

    If input shapes are not compatible

Source code in src/stimulus/utils/performance.py 
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
def handle_multiclass(
    self,
    labels: NDArray[np.float64],
    predictions: NDArray[np.float64],
) -> tuple[NDArray[np.float64], NDArray[np.float64]]:
    """Handle the case of multiclass classification.

    Args:
        labels: Labels array of shape (N,) or (N, 1)
        predictions: Predictions array of shape (N,) or (N, C) where C is number of classes

    Returns:
        tuple[NDArray[np.float64], NDArray[np.float64]]: Processed labels and predictions

    Raises:
        ValueError: If input shapes are not compatible
    """
    # Case 1: If labels are 2D with shape (N,1), squeeze to 1D shape (N,)
    # This handles cases where labels come as column vectors
    if len(labels.shape) == NON_SQUEEZED_SHAPE_LENGTH and labels.shape[1] == 1:
        labels = labels.squeeze(-1)

    if len(predictions.shape) == NON_SQUEEZED_SHAPE_LENGTH:
        # Case 2: Binary classification with shape (N,2)
        # Take probability of positive class (second column)
        if predictions.shape[1] == BINARY_CLASS_COUNT:
            predictions = predictions[:, 1]  # Shape becomes (N,)
            return labels, predictions
        # Case 3: Multi-class classification with shape (N,C)
        # Keep predictions as-is if labels are 1D and batch sizes match
        if len(labels.shape) == 1 and predictions.shape[0] == labels.shape[0]:
            return labels, predictions

    # If we get here, the shapes are not compatible
    raise ValueError(
        f"Incompatible shapes: labels {labels.shape}, predictions {predictions.shape}. "
        "Expected labels (N,) or (N, 1) and predictions (N,) or (N, C) where C is number of classes.",
    )

mcc 

mcc(
    labels: NDArray[float64], predictions: NDArray[float64]
) -> float

Compute Matthews Correlation Coefficient.

Source code in src/stimulus/utils/performance.py 
133
134
135
136
def mcc(self, labels: NDArray[np.float64], predictions: NDArray[np.float64]) -> float:
    """Compute Matthews Correlation Coefficient."""
    predictions_binary = np.array([1 if p > BINARY_THRESHOLD else 0 for p in predictions])
    return float(matthews_corrcoef(labels, predictions_binary))

prauc 

prauc(
    labels: NDArray[float64], predictions: NDArray[float64]
) -> float

Compute PR AUC score.

Source code in src/stimulus/utils/performance.py 
129
130
131
def prauc(self, labels: NDArray[np.float64], predictions: NDArray[np.float64]) -> float:
    """Compute PR AUC score."""
    return float(average_precision_score(labels, predictions))

precision 

precision(
    labels: NDArray[float64], predictions: NDArray[float64]
) -> float

Compute precision score.

Source code in src/stimulus/utils/performance.py 
143
144
145
146
def precision(self, labels: NDArray[np.float64], predictions: NDArray[np.float64]) -> float:
    """Compute precision score."""
    predictions_binary = np.array([1 if p > BINARY_THRESHOLD else 0 for p in predictions])
    return float(precision_score(labels, predictions_binary))

recall 

recall(
    labels: NDArray[float64], predictions: NDArray[float64]
) -> float

Compute recall score.

Source code in src/stimulus/utils/performance.py 
148
149
150
151
def recall(self, labels: NDArray[np.float64], predictions: NDArray[np.float64]) -> float:
    """Compute recall score."""
    predictions_binary = np.array([1 if p > BINARY_THRESHOLD else 0 for p in predictions])
    return float(recall_score(labels, predictions_binary))

rocauc 

rocauc(
    labels: NDArray[float64], predictions: NDArray[float64]
) -> float

Compute ROC AUC score.

Source code in src/stimulus/utils/performance.py 
125
126
127
def rocauc(self, labels: NDArray[np.float64], predictions: NDArray[np.float64]) -> float:
    """Compute ROC AUC score."""
    return float(roc_auc_score(labels, predictions))

spearmanr 

spearmanr(
    labels: NDArray[float64], predictions: NDArray[float64]
) -> float

Compute Spearman correlation coefficient.

Source code in src/stimulus/utils/performance.py 
153
154
155
def spearmanr(self, labels: NDArray[np.float64], predictions: NDArray[np.float64]) -> float:
    """Compute Spearman correlation coefficient."""
    return float(spearmanr(labels, predictions)[0])