Merge pull request #492 from Far0n/patch-1

bugfix evals_result regex
This commit is contained in:
Tianqi Chen 2015-09-14 08:46:58 -07:00
commit bda3282f6d

View File

@ -118,7 +118,7 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
sys.stderr.write(msg + '\n')
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):
evals_result[key].append(val)