fix(Evaluation): fix incorrect AUC computation - closes #165

frankkramer-lab · Jul 27, 2022 · 976369a · 976369a
1 parent 168c1e2
commit 976369a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aucmedi/evaluation/metrics.py b/aucmedi/evaluation/metrics.py
@@ -55,10 +55,10 @@ def compute_metrics(preds, labels, n_labels, threshold=None):
         if threshold is None:
             pred_argmax = np.argmax(preds, axis=-1)
             pred = (pred_argmax == c).astype(np.int)
-            pred_prob = np.max(preds, axis=-1)
         else:
             pred = np.where(preds[:, c] >= threshold, 1, 0)
-            pred_prob = preds[:, c]
+        # Obtain prediction confidence (probability)
+        pred_prob = preds[:, c]
 
         # Compute the confusion matrix
         tp, tn, fp, fn = compute_CM(truth, pred)