fix accuracy metric for lightgbm (#728)

mljar · Jun 3, 2024 · 1e2c26a · 1e2c26a
1 parent d2a8d3a
commit 1e2c26a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -14,7 +14,7 @@ seaborn>=0.11.1
 wordcloud>=1.8.1
 category_encoders>=2.2.2
 optuna>=2.7.0
-mljar-scikit-plot>=0.3.8
+mljar-scikit-plot>=0.3.11
 markdown
 typing-extensions
 ipython
diff --git a/supervised/utils/metric.py b/supervised/utils/metric.py
@@ -225,12 +225,6 @@ def lightgbm_eval_metric_accuracy(preds, dtrain):
     target = dtrain.get_label()
     weight = dtrain.get_weight()
 
-    unique_targets = np.unique(target)
-    if len(unique_targets) > 2:
-        cols = len(unique_targets)
-        rows = int(preds.shape[0] / len(unique_targets))
-        preds = np.reshape(preds, (rows, cols), order="F")
-
     return "accuracy", -negative_accuracy(target, preds, weight), True