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

Update xgboost.py

Browse Source
This commit is contained in:
Tianqi Chen 2015-05-13 23:15:19 -07:00
parent e03ef41829
commit 49ad633530
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
wrapper/xgboost.py
View File

@ -694,11 +694,11 @@ def train(params, dtrain, num_boost_round=10, evals=(), obj=None, feval=None,
sys.stderr.write("Stopping. Best iteration:\n{}\n\n".format(best_msg)) sys.stderr.write("Stopping. Best iteration:\n{}\n\n".format(best_msg))
bst.best_score = best_score bst.best_score = best_score
bst.best_iteration = best_score_i bst.best_iteration = best_score_i
return bst break
bst.best_score = best_score
bst.best_iteration = best_score_i
return bst return bst
class CVPack(object): class CVPack(object):
def __init__(self, dtrain, dtest, param): def __init__(self, dtrain, dtest, param):
self.dtrain = dtrain self.dtrain = dtrain