diff --git a/mapie/regression/quantile_regression.py b/mapie/regression/quantile_regression.py index 593a2a357..8c5ef3103 100644 --- 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,