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

23
plugin/updater_gpu/benchmark/benchmark.py
View File

@@ -14,19 +14,18 @@ def run_benchmark(args, gpu_algorithm, cpu_algorithm):
dtrain = xgb.DMatrix(X, y)
param = {'objective': 'binary:logistic',
'tree_method': 'exact',
'max_depth': 6,
'silent': 1,
'eval_metric': 'auc'}
param['updater'] = gpu_algorithm
print("Training with '%s'" % param['updater'])
param['tree_method'] = gpu_algorithm
print("Training with '%s'" % param['tree_method'])
tmp = time.time()
xgb.train(param, dtrain, args.iterations)
print ("Time: %s seconds" % (str(time.time() - tmp)))
param['updater'] = cpu_algorithm
print("Training with '%s'" % param['updater'])
param['tree_method'] = cpu_algorithm
print("Training with '%s'" % param['tree_method'])
tmp = time.time()
xgb.train(param, dtrain, args.iterations)
print ("Time: %s seconds" % (str(time.time() - tmp)))
@@ -34,17 +33,17 @@ def run_benchmark(args, gpu_algorithm, cpu_algorithm):
parser = argparse.ArgumentParser()
parser.add_argument('--algorithm', choices=['all', 'grow_gpu', 'grow_gpu_hist'], required=True)
parser.add_argument('--algorithm', choices=['all', 'gpu_exact', 'gpu_hist'], default='all')
parser.add_argument('--rows',type=int,default=1000000)
parser.add_argument('--columns',type=int,default=50)
parser.add_argument('--iterations',type=int,default=500)
args = parser.parse_args()
if 'grow_gpu_hist' in args.algorithm:
run_benchmark(args, args.algorithm, 'grow_fast_histmaker')
if 'grow_gpu' in args.algorithm:
run_benchmark(args, args.algorithm, 'grow_colmaker')
if 'gpu_hist' in args.algorithm:
run_benchmark(args, args.algorithm, 'hist')
if 'gpu_exact' in args.algorithm:
run_benchmark(args, args.algorithm, 'exact')
if 'all' in args.algorithm:
run_benchmark(args, 'grow_gpu', 'grow_colmaker')
run_benchmark(args, 'grow_gpu_hist', 'grow_fast_histmaker')
run_benchmark(args, 'gpu_exact', 'exact')
run_benchmark(args, 'gpu_hist', 'hist')