Merge pull request scikit-learn#2753 from pprett/gbrt-verbose-remaining-time

pprett · pprett · commit 1c6c36bff9a3 · 2014-01-14T02:56:32.000-08:00
fix: remaining time not currectly computed
diff --git a/sklearn/ensemble/gradient_boosting.py b/sklearn/ensemble/gradient_boosting.py
@@ -499,7 +499,7 @@ def update(self, j, est):
         i = j - self.begin_at_stage  # iteration relative to the start iter
         if (i + 1) % self.verbose_mod == 0:
             oob_impr = est.oob_improvement_[j] if do_oob else 0
-            remaining_time = ((est.n_estimators - (i + 1)) *
+            remaining_time = ((est.n_estimators - (j + 1)) *
                               (time() - self.start_time) / float(i + 1))
             if remaining_time > 60:
                 remaining_time = '{0:.2f}m'.format(remaining_time / 60.0)
