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

correct format

Browse Source
This commit is contained in:
Yizheng Liao 2015-04-23 21:27:12 -07:00
parent 44d1043031
commit 1d8fc6280c
1 changed files with 2 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
wrapper/xgboost.py
View File

@ -531,7 +531,8 @@ class Booster(object):
return fmap
def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None, early_stopping_rounds=None,evals_result=None):
def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
early_stopping_rounds=None,evals_result=None):
"""
Train a booster with given parameters.
@ -584,18 +585,8 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None, ea
bst_eval_set = bst.eval_set(evals, i, feval)
if isinstance(bst_eval_set, string_types):
msg = bst_eval_set
#sys.stderr.write(bst_eval_set + '\n')
# if evals_result is not None:
# res = re.findall(":([0-9.]+).",bst_eval_set)
# for key,val in zip(evals_name,res):
# evals_result[key].append(val)
else:
msg = bst_eval_set.decode()
# sys.stderr.write(bst_eval_set.decode() + '\n')
# if evals_result is not None:
# res = re.findall(":([0-9.]+).",bst_eval_set.decode())
# for key,val in zip(evals_name,res):
# evals_result[key].append(val)
sys.stderr.write(msg + '\n')
if evals_result is not None: