Skip to content

Commit

Permalink
Update xgboost.py
Browse files Browse the repository at this point in the history
tqchen committed May 14, 2015
1 parent e03ef41 commit 49ad633
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions wrapper/xgboost.py
Original file line number Diff line number Diff line change
@@ -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))
bst.best_score = best_score
bst.best_iteration = best_score_i
return bst

break
bst.best_score = best_score
bst.best_iteration = best_score_i
return bst


class CVPack(object):
def __init__(self, dtrain, dtest, param):
self.dtrain = dtrain

0 comments on commit 49ad633

Please sign in to comment.