fit base estimator after calibration

Prokopios Gryllos · Prokopios Gryllos · commit 132864b36154 · 2017-11-12T17:53:07.000+01:00
diff --git a/sklearn/calibration.py b/sklearn/calibration.py
@@ -98,6 +98,7 @@ def fit(self, X, y):
                 )
             self.calibrated_threshold = sum(calibrated_thresholds) /\
                 len(calibrated_thresholds)
+            self.base_estimator.fit(X, y)
         return self
 
     def predict(self, X):

Original file line number	Diff line number	Diff line change
`@@ -98,6 +98,7 @@ def fit(self, X, y):`
`98`	`98`	`)`
`99`	`99`	`self.calibrated_threshold = sum(calibrated_thresholds) /\`
`100`	`100`	`len(calibrated_thresholds)`
	`101`	`+ self.base_estimator.fit(X, y)`
`101`	`102`	`return self`
`102`	`103`
`103`	`104`	`def predict(self, X):`