debug test

inria-thoth · Sep 30, 2024 · 794bfe8 · 794bfe8
1 parent 1b970b5
commit 794bfe8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cyanure/estimators.py b/cyanure/estimators.py
@@ -287,7 +287,7 @@ def fit(self, X, labels, le_parameter=None):
 
         if (self.multi_class == "multinomial" or
            (self.multi_class == "auto" and not self._binary_problem)) and self.loss == "logistic":
-            if len(np.unique(labels)) != 2 or self.multi_class == "multinomial":
+            if len(np.unique(labels)) != 2:
                 self._binary_problem = False
 
                 loss = "multiclass-logistic"