BUG Fixes bug

thomasjpfan · thomasjpfan · commit 5c22dec68d74 · 2021-11-18T08:59:58.000-08:00
diff --git a/sklearn/ensemble/_gb.py b/sklearn/ensemble/_gb.py
@@ -736,7 +736,7 @@ def _raw_predict(self, X):
         predict_stages(self.estimators_, X, self.learning_rate, raw_predictions)
         return raw_predictions
 
-    def _staged_raw_predict(self, X, check_input=False):
+    def _staged_raw_predict(self, X, check_input=True):
         """Compute raw predictions of ``X`` for each iteration.
 
         This method allows monitoring (i.e. determine error on testing set)
