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

Fix demo typo (#2632)

Browse Source
This commit is contained in:
Rory Mitchell 2017-08-24 03:21:51 +12:00 committed by Michaël Benesty
parent cd366ecb4b
commit 70071fc38c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
demo/gpu_acceleration/cover_type.py
View File

@ -29,12 +29,12 @@ dtest = xgb.DMatrix(X_test, label=y_test)
gpu_res = {} # Store accuracy result
tmp = time.time()
# Train model
xgb.train(param, dtrain, numround, evals=[(dtest, 'test')], evals_result=gpu_res)
xgb.train(param, dtrain, num_round, evals=[(dtest, 'test')], evals_result=gpu_res)
print("GPU Training Time: %s seconds" % (str(time.time() - tmp)))
# Repeat for CPU algorithm
tmp = time.time()
param['tree_method'] = 'hist'
cpu_res = {}
xgb.train(param, dtrain, numround, evals=[(dtest, 'test')], evals_result=cpu_res)
xgb.train(param, dtrain, num_round, evals=[(dtest, 'test')], evals_result=cpu_res)
print("CPU Training Time: %s seconds" % (str(time.time() - tmp)))