estimator instead of base_estimator

GauravPandeyLab · Dec 13, 2023 · 95a3c17 · 95a3c17
1 parent c37b448
commit 95a3c17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/eipy/ei.py b/eipy/ei.py
@@ -570,7 +570,7 @@ def _train_predict_single_base_predictor(
  )
 
  if self.calibration_model is not None:
- self.calibration_model.base_estimator = model
+ self.calibration_model.estimator = model
  model = self.calibration_model
 
  model.fit(X_sample, y_sample)