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

ENH: Allow possibly negative evaluation metrics.

Browse Source
This commit is contained in:
Skipper Seabold 2015-05-20 14:17:30 -05:00
parent 0f5f9c0385
commit 3952b525b8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wrapper/xgboost.py
View File

@ -795,7 +795,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)
return bst return bst
@ -842,7 +842,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)