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

12
tests/python-gpu/test_gpu_updaters.py
View File

@@ -20,16 +20,6 @@ datasets = ["Boston", "Cancer", "Digits", "Sparse regression",
class TestGPU(unittest.TestCase):
def test_gpu_exact(self):
variable_param = {'max_depth': [2, 6, 15], }
for param in parameter_combinations(variable_param):
param['tree_method'] = 'gpu_exact'
gpu_results = run_suite(param, select_datasets=datasets)
assert_results_non_increasing(gpu_results, 1e-2)
param['tree_method'] = 'exact'
cpu_results = run_suite(param, select_datasets=datasets)
assert_gpu_results(cpu_results, gpu_results)
def test_gpu_hist(self):
test_param = parameter_combinations({'n_gpus': [1], 'max_depth': [2, 8],
'max_leaves': [255, 4],
@@ -65,7 +55,7 @@ class TestGPU(unittest.TestCase):
'max_leaves': [255, 4],
'max_bin': [2, 64],
'grow_policy': ['lossguide'],
'tree_method': ['gpu_hist', 'gpu_exact']}
'tree_method': ['gpu_hist']}
for param in parameter_combinations(variable_param):
gpu_results = run_suite(param, select_datasets=datasets)
assert_results_non_increasing(gpu_results, 1e-2)