Merge pull request dmlc#492 from Far0n/patch-1

bugfix evals_result regex
sysucjl · Sep 14, 2015 · bda3282 · bda3282
2 parents b1c94c7 + 0406c64
commit bda3282
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-package/xgboost/training.py b/python-package/xgboost/training.py
@@ -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)