Update eval_metric

ThomasMeissnerDS · Sep 13, 2023 · cf0e9a0 · cf0e9a0
1 parent 198c726
commit cf0e9a0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/bluecast/evaluation/eval_metrics.py b/bluecast/evaluation/eval_metrics.py
@@ -56,12 +56,14 @@ def eval_classifier(
         logger(f"The balanced logloss is {bll}")
     else:
         bll = 99
-        logger(f"Skip blanced logloss as number of classes is {pd.Series(y_classes).nunique()}.")
+        logger(
+            f"Skip blanced logloss as number of classes is {pd.Series(y_classes).nunique()}."
+        )
 
     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')
+        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}")

diff --git a/dist/bluecast-0.21-py3-none-any.whl b/dist/bluecast-0.21-py3-none-any.whl
diff --git a/dist/bluecast-0.21.tar.gz b/dist/bluecast-0.21.tar.gz