TST: Added test for early stopping

2015-10-04 23:15:25 -05:00
parent 412310ed04
commit 956e50686e
2 changed files with 11 additions and 4 deletions
--- a/tests/python/test_with_sklearn.py
+++ b/tests/python/test_with_sklearn.py
@@ -1,9 +1,7 @@
-import pickle
 import xgboost as xgb
-
 import numpy as np
 from sklearn.cross_validation import KFold, train_test_split
-from sklearn.metrics import confusion_matrix, mean_squared_error
+from sklearn.metrics import mean_squared_error
 from sklearn.grid_search import GridSearchCV
 from sklearn.datasets import load_iris, load_digits, load_boston

@@ -45,4 +43,4 @@ def test_boston_housing_regression():
 	assert mean_squared_error(preds, labels) < 9


-test_boston_housing_regression()
+