add accuracy eval_metric (#362)

mljar · Apr 1, 2021 · fae708d · fae708d
1 parent ad49896
commit fae708d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/supervised/utils/metric.py b/supervised/utils/metric.py
@@ -57,7 +57,7 @@ def negative_f1(y_true, y_predicted, sample_weight=None):
         y_predicted = y_predicted.ravel()
 
     average = None
-    if len(y_predicted.shape) == 1 or (len(y_predicted.shape) == 2 and y_predicted.shape[1] == 1):
+    if len(y_predicted.shape) == 1:
         y_predicted = (y_predicted > 0.5).astype(int)
         average = "binary"
     else: