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

bugfix evals_result regex

Browse Source
This commit is contained in:
Far0n 2015-09-14 11:25:41 +02:00
parent b1c94c7d86
commit 0406c64a5d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -118,7 +118,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
sys.stderr.write(msg + '\n') sys.stderr.write(msg + '\n')
if evals_result is not None: if evals_result is not None:
res = re.findall(":-([0-9.]+).", msg) res = re.findall(":-?([0-9.]+).", msg)
for key, val in zip(evals_name, res): for key, val in zip(evals_name, res):
evals_result[key].append(val) evals_result[key].append(val)