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

Merge pull request #644 from Far0n/verbose_eval_patch

Browse Source 
small verbose_eval fixes
This commit is contained in:
Yuan (Terry) Tang 2015-12-01 14:58:58 -06:00
commit a1c0ee0e66
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: else:
if isinstance(verbose_eval, int): if isinstance(verbose_eval, int):
verbose_eval_every_line = verbose_eval 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 xgb_model is not None:
if not isinstance(xgb_model, STRING_TYPES): 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:
if verbose_eval_every_line: 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') sys.stderr.write(msg + '\n')
else: else:
sys.stderr.write(msg + '\n') sys.stderr.write(msg + '\n')