fix(Evaluation): updated deprecated np.int to np.int8 for boolean to …

…integer matrix conversion
frankkramer-lab · Dec 20, 2022 · 8f1a2e8 · 8f1a2e8
1 parent 973ff46
commit 8f1a2e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aucmedi/evaluation/metrics.py b/aucmedi/evaluation/metrics.py
@@ -54,7 +54,7 @@ def compute_metrics(preds, labels, n_labels, threshold=None):
         truth = labels[:, c]
         if threshold is None:
             pred_argmax = np.argmax(preds, axis=-1)
-            pred = (pred_argmax == c).astype(np.int)
+            pred = (pred_argmax == c).astype(np.int8)
         else:
             pred = np.where(preds[:, c] >= threshold, 1, 0)
         # Obtain prediction confidence (probability)