From 70071fc38c3a93966ce276ac609cfbdf96f89526 Mon Sep 17 00:00:00 2001 From: Rory Mitchell Date: Thu, 24 Aug 2017 03:21:51 +1200 Subject: [PATCH] Fix demo typo (#2632) --- demo/gpu_acceleration/cover_type.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/demo/gpu_acceleration/cover_type.py b/demo/gpu_acceleration/cover_type.py index 5183cd7e7..5a073eb2c 100644 --- a/demo/gpu_acceleration/cover_type.py +++ b/demo/gpu_acceleration/cover_type.py @@ -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)))