10000 fix: remaining time not currectly computed · mohitsingh1007/scikit-learn@bac878f · GitHub
[go: up one dir, main page]

Skip to content

Commit bac878f

Browse files
committed
fix: remaining time not currectly computed
1 parent bc6ad64 commit bac878f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sklearn/ensemble/gradient_boosting.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -499,7 +499,7 @@ def update(self, j, est):
499499
i = j - self.begin_at_stage # iteration relative to the start iter
500500
if (i + 1) % self.verbose_mod == 0:
501501
oob_impr = est.oob_improvement_[j] if do_oob else 0
502-
remaining_time = ((est.n_estimators - (i + 1)) *
502+
remaining_time = ((est.n_estimators - (j + 1)) *
503503
(time() - self.start_time) / float(i + 1))
504504
if remaining_time > 60:
505505
remaining_time = '{0:.2f}m'.format(remaining_time / 60.0)

0 commit comments

Comments
 (0)
0