Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added fixed random seed for tests (+1 squashed commit)

Browse Source 
Squashed commits:
[76e3664] Added fixed random seed for tests
This commit is contained in:
terrytangyuan
2015-10-21 23:24:37 -05:00
parent 755072e378
commit ec2cdafec5
3 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/python/test_early_stopping.py
View File

@@ -1,14 +1,19 @@
import xgboost as xgb
import numpy as np
from sklearn.datasets import load_digits
from sklearn.cross_validation import KFold, train_test_split
rng = np.random.RandomState(1994)
def test_early_stopping_nonparallel():
digits = load_digits(2)
X = digits['data']
y = digits['target']
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=0)
clf = xgb.XGBClassifier()
clf.fit(X_train, y_train, early_stopping_rounds=10, eval_metric="auc",
eval_set=[(X_test, y_test)])
# digits = load_digits(2)
# X = digits['data']
# y = digits['target']
# X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=0)
# clf = xgb.XGBClassifier()
# clf.fit(X_train, y_train, early_stopping_rounds=10, eval_metric="auc",
# eval_set=[(X_test, y_test)])
print("This test will be re-visited later. ")
# TODO: parallel test for early stopping
# TODO: comment out for now. Will re-visit later