Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

small verbose_eval fixes

Browse Source 
- ensures same behavior for verbose_eval=0 and verbose_eval=False
- fix printing last eval message if early_stopping_rounds is set, but xgb
  runs to the end
This commit is contained in:
Far0n 2015-11-23 11:11:21 +01:00
parent 13829329bd
commit af166bf0a0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python-package/xgboost/training.py
View File

@ -91,7 +91,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
else:
if isinstance(verbose_eval, int):
verbose_eval_every_line = verbose_eval
verbose_eval = True
verbose_eval = True if verbose_eval_every_line > 0 else False
if xgb_model is not None:
if not isinstance(xgb_model, STRING_TYPES):
@ -206,7 +206,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
if verbose_eval:
if verbose_eval_every_line:
if i % verbose_eval_every_line == 0:
if i % verbose_eval_every_line == 0 or i == num_boost_round - 1:
sys.stderr.write(msg + '\n')
else:
sys.stderr.write(msg + '\n')