debug ci

inria-thoth · Sep 27, 2024 · 585e436 · 585e436
1 parent 47f30e5
commit 585e436
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/cyanure/estimators.py b/cyanure/estimators.py
@@ -259,6 +259,11 @@ def __init__(self, loss='square', penalty='l2', fit_intercept=False, dual=None,
         self.n_threads = n_threads
         self.safe = safe
 
+        if (loss == "multiclass-logistic" or loss == "logistic") and self.lambda_1 == -10.0:
+            self.lambda_1 = 0
+        elif (self.lambda_1 == -10.0):
+            self.lambda_1 = 0
+
     def fit(self, X, labels, le_parameter=None):
         """
         Fit the parameters.
@@ -298,11 +303,6 @@ def fit(self, X, labels, le_parameter=None):
         if loss is None:
             loss = self.loss
 
-        if (loss == "multiclass-logistic" or loss == "logistic") and self.lambda_1 == -10.0:
-            self.lambda_1 = 0
-        elif (self.lambda_1 == -10.0):
-            self.lambda_1 = 0
-
         labels = np.squeeze(labels)
         initial_weight, yf, nclasses = self._initialize_weight(X, labels)