From b3f3e7d0cbde1877359d9e9015e9e63db4880e69 Mon Sep 17 00:00:00 2001 From: Faron Date: Sat, 19 Dec 2015 10:35:16 +0100 Subject: [PATCH] fixed wrong iter when using training continuation --- python-package/xgboost/training.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python-package/xgboost/training.py b/python-package/xgboost/training.py index e47db0bc6..d0fc1363c 100644 --- a/python-package/xgboost/training.py +++ b/python-package/xgboost/training.py @@ -117,7 +117,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None, evals_result.update(dict([(key, {}) for key in evals_name])) if not early_stopping_rounds: - for i in range(num_boost_round): + for i in range(nboost, nboost + num_boost_round): bst.update(dtrain, i, obj) nboost += 1 if len(evals) != 0: @@ -189,7 +189,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None, if isinstance(learning_rates, list) and len(learning_rates) != num_boost_round: raise ValueError("Length of list 'learning_rates' has to equal 'num_boost_round'.") - for i in range(num_boost_round): + for i in range(nboost, nboost + num_boost_round): if learning_rates is not None: if isinstance(learning_rates, list): bst.set_param({'eta': learning_rates[i]})