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

bugfix: when metric's name contains - (#2090)

Browse Source 
When metric's name contains `-`, Python will complain about insufficient arguments to unpack.
This commit is contained in:
Liam Huang 2017-03-17 01:36:39 +08:00 committed by Tianqi Chen
parent fee1181803
commit 3a2b8332a6
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -79,7 +79,9 @@ def record_evaluation(eval_result):
def init(env): def init(env):
"""internal function""" """internal function"""
for k, _ in env.evaluation_result_list: for k, _ in env.evaluation_result_list:
key, metric = k.split('-') pos = k.index('-')
key = k[:pos]
metric = k[pos + 1:]
if key not in eval_result: if key not in eval_result:
eval_result[key] = {} eval_result[key] = {}
if metric not in eval_result[key]: if metric not in eval_result[key]:
@ -90,7 +92,9 @@ def record_evaluation(eval_result):
if len(eval_result) == 0: if len(eval_result) == 0:
init(env) init(env)
for k, v in env.evaluation_result_list: for k, v in env.evaluation_result_list:
key, metric = k.split('-') pos = k.index('-')
key = k[:pos]
metric = k[pos + 1:]
eval_result[key][metric].append(v) eval_result[key][metric].append(v)
return callback return callback