Update eval_metric

ThomasMeissnerDS · Sep 13, 2023 · cd4ab0a · cd4ab0a
1 parent ddde588
commit cd4ab0a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bluecast/evaluation/eval_metrics.py b/bluecast/evaluation/eval_metrics.py
@@ -58,7 +58,10 @@ def eval_classifier(
         logger(f"Skip blanced logloss as number of classes is {pd.Series(y_classes).nunique()}.")
         logloss = 99
 
-    roc_auc = roc_auc_score(y_true, y_probs)
+    if pd.Series(y_classes).nunique() <= 2:
+        roc_auc = roc_auc_score(y_true, y_probs)
+    else:
+        roc_auc = roc_auc_score(y_true, y_probs, multi_class='ovr')
     logger(f"The ROC auc score is {roc_auc}")
     logloss = log_loss(y_true, y_probs)
     logger(f"The log loss score is {logloss}")