Update xgboost.py

2015-05-13 23:15:19 -07:00 · 2015-05-13 23:15:19 -07:00 · 49ad633530
commit 49ad633530
parent e03ef41829
1 changed files with 3 additions and 3 deletions
--- a/wrapper/xgboost.py
+++ b/wrapper/xgboost.py
@ -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