diff --git a/python-package/xgboost/training.py b/python-package/xgboost/training.py index ee189e88d..b2cb54d9a 100644 --- a/python-package/xgboost/training.py +++ b/python-package/xgboost/training.py @@ -39,8 +39,8 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None, If there's more than one, will use the last. Returns the model from the last iteration (not the best one). If early stopping occurs, the model will have three additional fields: - bst.best_score, bst.best_iteration and bst.best_ntree_limit. - (Use bst.best_ntree_limit to get the correct value if num_parallel_tree + bst.best_score, bst.best_iteration and bst.best_ntree_limit. + (Use bst.best_ntree_limit to get the correct value if num_parallel_tree and/or num_class appears in the parameters) evals_result: dict This dictionary stores the evaluation results of all the items in watchlist.