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

[GPU-Plugin] Change GPU plugin to use tree_method parameter, bump cmake version to 3.5 for GPU plugin, add compute architecture 3.5, remove unused cmake files (#2455)

Browse Source
This commit is contained in:
Rory Mitchell
2017-06-29 16:19:45 +12:00
committed by GitHub
parent 88488fdbb9
commit 48f3003302
8 changed files with 168 additions and 835 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
plugin/updater_gpu/test/python/test.py
View File

@@ -35,7 +35,7 @@ class TestGPU(unittest.TestCase):
'objective': 'binary:logistic',
'eval_metric': 'auc'}
ag_param2 = {'max_depth': 2,
'updater': 'grow_gpu',
'tree_method': 'gpu_exact',
'eta': 1,
'silent': 1,
'objective': 'binary:logistic',
@@ -59,7 +59,7 @@ class TestGPU(unittest.TestCase):
dtest = xgb.DMatrix(X_test, y_test)
param = {'objective': 'binary:logistic',
'updater': 'grow_gpu',
'tree_method': 'gpu_exact',
'max_depth': 3,
'eval_metric': 'auc'}
res = {}
@@ -75,7 +75,7 @@ class TestGPU(unittest.TestCase):
dtrain2 = xgb.DMatrix(X2, label=y2)
param = {'objective': 'binary:logistic',
'updater': 'grow_gpu',
'tree_method': 'gpu_exact',
'max_depth': 2,
'eval_metric': 'auc'}
res = {}
@@ -134,7 +134,7 @@ class TestGPU(unittest.TestCase):
'objective': 'binary:logistic',
'eval_metric': 'auc'}
ag_param2 = {'max_depth': max_depth,
'updater': 'grow_gpu_hist',
'tree_method': 'gpu_hist',
'eta': 1,
'silent': 1,
'n_gpus': 1,
@@ -142,7 +142,7 @@ class TestGPU(unittest.TestCase):
'max_bin': max_bin,
'eval_metric': 'auc'}
ag_param3 = {'max_depth': max_depth,
'updater': 'grow_gpu_hist',
'tree_method': 'gpu_hist',
'eta': 1,
'silent': 1,
'n_gpus': n_gpus,
@@ -177,7 +177,7 @@ class TestGPU(unittest.TestCase):
dtest = xgb.DMatrix(X_test, y_test)
param = {'objective': 'binary:logistic',
'updater': 'grow_gpu_hist',
'tree_method': 'gpu_hist',
'max_depth': max_depth,
'n_gpus': 1,
'max_bin': max_bin,
@@ -189,7 +189,7 @@ class TestGPU(unittest.TestCase):
assert self.non_decreasing(res['train']['auc'])
#assert self.non_decreasing(res['test']['auc'])
param2 = {'objective': 'binary:logistic',
'updater': 'grow_gpu_hist',
'tree_method': 'gpu_hist',
'max_depth': max_depth,
'n_gpus': n_gpus,
'max_bin': max_bin,
@@ -211,7 +211,7 @@ class TestGPU(unittest.TestCase):
dtrain2 = xgb.DMatrix(X2, label=y2)
param = {'objective': 'binary:logistic',
'updater': 'grow_gpu_hist',
'tree_method': 'gpu_hist',
'max_depth': max_depth,
'n_gpus': n_gpus,
'max_bin': max_bin,
@@ -250,7 +250,7 @@ class TestGPU(unittest.TestCase):
######################################################################
# fail-safe test for max_bin
param = {'objective': 'binary:logistic',
'updater': 'grow_gpu_hist',
'tree_method': 'gpu_hist',
'max_depth': max_depth,
'n_gpus': n_gpus,
'eval_metric': 'auc',
@@ -263,7 +263,7 @@ class TestGPU(unittest.TestCase):
######################################################################
# subsampling
param = {'objective': 'binary:logistic',
'updater': 'grow_gpu_hist',
'tree_method': 'gpu_hist',
'max_depth': max_depth,
'n_gpus': n_gpus,
'eval_metric': 'auc',
@@ -279,7 +279,7 @@ class TestGPU(unittest.TestCase):
######################################################################
# fail-safe test for max_bin=2
param = {'objective': 'binary:logistic',
'updater': 'grow_gpu_hist',
'tree_method': 'gpu_hist',
'max_depth': 2,
'n_gpus': n_gpus,
'eval_metric': 'auc',