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

Update training.py

Browse Source
This commit is contained in:
quansie 2015-10-12 14:19:25 +02:00
parent 8a484e990e
commit 541580d157
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -73,12 +73,12 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
if evals_result is not None: if evals_result is not None:
res = re.findall("([0-9a-zA-Z@]+[-]*):-?([0-9.]+).", msg) res = re.findall("([0-9a-zA-Z@]+[-]*):-?([0-9.]+).", msg)
for key in evals_name: for key in evals_name:
evals_idx = evals_name.index(key) evals_idx = evals_name.index(key)
res_per_eval = len(res) / len(evals_name) res_per_eval = len(res) / len(evals_name)
for r in range(res_per_eval): for r in range(res_per_eval):
res_item = res[(evals_idx*res_per_eval) + r] res_item = res[(evals_idx*res_per_eval) + r]
res_key = res_item[0] res_key = res_item[0]
res_val = res_item[1] res_val = res_item[1]
if res_key in evals_result[key]: if res_key in evals_result[key]:
evals_result[key][res_key].append(res_val) evals_result[key][res_key].append(res_val)
else: else:
@ -130,12 +130,12 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
if evals_result is not None: if evals_result is not None:
res = re.findall("([0-9a-zA-Z@]+[-]*):-?([0-9.]+).", msg) res = re.findall("([0-9a-zA-Z@]+[-]*):-?([0-9.]+).", msg)
for key in evals_name: for key in evals_name:
evals_idx = evals_name.index(key) evals_idx = evals_name.index(key)
res_per_eval = len(res) / len(evals_name) res_per_eval = len(res) / len(evals_name)
for r in range(res_per_eval): for r in range(res_per_eval):
res_item = res[(evals_idx*res_per_eval) + r] res_item = res[(evals_idx*res_per_eval) + r]
res_key = res_item[0] res_key = res_item[0]
res_val = res_item[1] res_val = res_item[1]
if res_key in evals_result[key]: if res_key in evals_result[key]:
evals_result[key][res_key].append(res_val) evals_result[key][res_key].append(res_val)
else: else: