add init_fit

scikit-learn-contrib · Dec 16, 2024 · 1b89d83 · 1b89d83
1 parent 406b500
commit 1b89d83
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/mapie/regression/quantile_regression.py b/mapie/regression/quantile_regression.py
@@ -546,9 +546,8 @@ def fit(
         MapieQuantileRegressor
              The model itself.
         """
-        self.cv = self._check_cv(cast(str, self.cv))
-        self.alpha_np = self._check_alpha(self.alpha)
-        self.estimators_: List[RegressorMixin] = []
+
+        self.init_fit()
 
         if self.cv == "prefit":
             X_calib, y_calib = self.prefit_estimators(X, y)
@@ -571,6 +570,12 @@ def fit(
 
         return self
 
+    def init_fit(self):
+
+        self.cv = self._check_cv(cast(str, self.cv))
+        self.alpha_np = self._check_alpha(self.alpha)
+        self.estimators_: List[RegressorMixin] = []
+
     def prefit_estimators(
         self,
         X: ArrayLike,