Fix off-by-one error for best iteration calculation (closes #6014) (#6016)

This commit is contained in:
Marek Grzenkowicz 2020-09-02 15:15:45 +02:00 committed by GitHub
parent f7a25d69f7
commit 92d7832a86
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -554,9 +554,10 @@ def train(
iter_since_best = 0 iter_since_best = 0
best_score = current_score best_score = current_score
if iter_since_best >= n_early_stopping: if iter_since_best >= n_early_stopping:
iter_current = i + 1
msg.text( msg.text(
"Early stopping, best iteration " "Early stopping, best iteration "
"is: {}".format(i - iter_since_best) "is: {}".format(iter_current - iter_since_best)
) )
msg.text( msg.text(
"Best score = {}; Final iteration " "Best score = {}; Final iteration "