staged_predict predicts classes, not probabilites

abhishek thakur · abhishek thakur · commit 1d0da8a5bdd3 · 2014-03-13T17:23:24.000+01:00
diff --git a/sklearn/ensemble/gradient_boosting.py b/sklearn/ensemble/gradient_boosting.py
@@ -1186,7 +1186,7 @@ def predict(self, X):
         return self.classes_.take(np.argmax(proba, axis=1), axis=0)
 
     def staged_predict(self, X):
-        """Predict class probabilities at each stage for X.
+        """Predict classes at each stage for X.
 
         This method allows monitoring (i.e. determine error on testing set)
         after each stage.
