fix: classes parameter (np.array)

frankkramer-lab · Jan 18, 2024 · 78b3078 · 78b3078
1 parent 812bb88
commit 78b3078
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aucmedi/utils/class_weights.py b/aucmedi/utils/class_weights.py
@@ -100,7 +100,7 @@ def compute_multilabel_weights(ohe_array, method="balanced"):
     class_weights = np.empty([n_classes])
     # Compute weight for each class individually
     for i in range(0, n_classes):
-        weight = compute_class_weight(class_weight=method, classes=[0,1],
+        weight = compute_class_weight(class_weight=method, classes=np.array([0,1]),
                                       y=ohe_array[:, i])
         class_weights[i] = weight[1]
     # Return resulting class weight list