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

Remove gpu_exact tree method (#4742)

Browse Source
This commit is contained in:
Rong Ou
2019-08-06 16:43:20 -07:00
committed by Rory Mitchell
parent 2a4df8e29f
commit 851b5b3808
16 changed files with 36 additions and 971 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/python/test_with_sklearn.py
View File

@@ -444,9 +444,9 @@ def test_sklearn_n_jobs():
def test_kwargs():
params = {'updater': 'grow_gpu', 'subsample': .5, 'n_jobs': -1}
params = {'updater': 'grow_gpu_hist', 'subsample': .5, 'n_jobs': -1}
clf = xgb.XGBClassifier(n_estimators=1000, **params)
assert clf.get_params()['updater'] == 'grow_gpu'
assert clf.get_params()['updater'] == 'grow_gpu_hist'
assert clf.get_params()['subsample'] == .5
assert clf.get_params()['n_estimators'] == 1000
@@ -472,7 +472,7 @@ def test_kwargs_grid_search():
def test_kwargs_error():
params = {'updater': 'grow_gpu', 'subsample': .5, 'n_jobs': -1}
params = {'updater': 'grow_gpu_hist', 'subsample': .5, 'n_jobs': -1}
with pytest.raises(TypeError):
clf = xgb.XGBClassifier(n_jobs=1000, **params)
assert isinstance(clf, xgb.XGBClassifier)