diff --git a/cyanure/estimators.py b/cyanure/estimators.py index 3dd7038fa..c4a7359e3 100644 --- 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"