From 794bfe826d9c2298de9904f832e22977da549490 Mon Sep 17 00:00:00 2001 From: RYCKEBOER Thomas <37658138+thomasryck@users.noreply.github.com> Date: Mon, 30 Sep 2024 13:43:34 +0200 Subject: [PATCH] debug test --- cyanure/estimators.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"