Speed up python test (#5752)

* Speed up tests

* Prevent DeviceQuantileDMatrix initialisation with numpy

* Use joblib.memory

* Use RandomState
This commit is contained in:
Rory Mitchell
2020-06-05 11:39:24 +12:00
committed by GitHub
parent cfc23c6a6b
commit 359023c0fa
5 changed files with 53 additions and 18 deletions

View File

@@ -47,6 +47,7 @@ class TestGPU(unittest.TestCase):
device_dmatrix_datasets = ["Boston", "Cancer", "Digits"]
for param in test_param:
param['tree_method'] = 'gpu_hist'
gpu_results_device_dmatrix = run_suite(param, select_datasets=device_dmatrix_datasets,
DMatrixT=xgb.DeviceQuantileDMatrix,
dmatrix_params={'max_bin': param['max_bin']})