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

Speed up python test (#5752)

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files

4
python-package/xgboost/data.py
View File

@@ -566,10 +566,6 @@ class DeviceQuantileCudaArrayInterfaceHandler(
__device_quantile_dmatrix_registry.register_handler(
'cupy.core.core', 'ndarray', DeviceQuantileCudaArrayInterfaceHandler)
__device_quantile_dmatrix_registry.register_handler_opaque(
lambda x: hasattr(x, '__array__'), NumpyHandler)
__device_quantile_dmatrix_registry.register_handler_opaque(
lambda x: hasattr(x, '__cuda_array_interface__'), NumpyHandler)
class DeviceQuantileCudaColumnarHandler(DeviceQuantileDMatrixDataHandler,