Fix demo typo (#2632)

dmlc · Aug 23, 2017 · 70071fc · 70071fc
1 parent cd366ec
commit 70071fc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/demo/gpu_acceleration/cover_type.py b/demo/gpu_acceleration/cover_type.py
@@ -29,12 +29,12 @@
 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)))