fix test.

Alnusjaponica · Apr 17, 2017 · f6b25ac · f6b25ac
1 parent 05f4a26
commit f6b25ac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python_package_test/test_engine.py b/tests/python_package_test/test_engine.py
@@ -32,8 +32,8 @@ class template(object):
     @staticmethod
     def test_template(params={'objective': 'regression', 'metric': 'l2'},
                       X_y=load_boston(True), feval=mean_squared_error,
-                      num_round=200, init_model=None, custom_eval=None,
-                      early_stopping_rounds=5,
+                      num_round=50, init_model=None, custom_eval=None,
+                      early_stopping_rounds=2,
                       return_data=False, return_model=False):
         params['verbose'], params['seed'] = -1, 42
         X_train, X_test, y_train, y_test = train_test_split(*X_y, test_size=0.1, random_state=42)