Update Python documents. (#6376)

2020-11-12 17:51:32 +08:00
parent c5645180a6
commit c90f968d92
4 changed files with 18 additions and 11 deletions
--- a/tests/python/test_callback.py
+++ b/tests/python/test_callback.py
@@ -123,9 +123,10 @@ class TestCallbacks(unittest.TestCase):
        X, y = load_breast_cancer(return_X_y=True)
        cls = xgb.XGBClassifier()
        early_stopping_rounds = 5
+        early_stop = xgb.callback.EarlyStopping(rounds=early_stopping_rounds)
        cls.fit(X, y, eval_set=[(X, y)],
-                early_stopping_rounds=early_stopping_rounds,
-                eval_metric=tm.eval_error_metric)
+                eval_metric=tm.eval_error_metric,
+                callbacks=[early_stop])
        booster = cls.get_booster()
        dump = booster.get_dump(dump_format='json')
        assert len(dump) - booster.best_iteration == early_stopping_rounds + 1