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

9
tests/python-gpu/test_gpu_linear.py
View File

@@ -3,12 +3,11 @@ import pytest
import unittest
sys.path.append('tests/python/')
import test_linear # noqa: E402
import testing as tm # noqa: E402
import test_linear # noqa: E402
import testing as tm # noqa: E402
class TestGPULinear(unittest.TestCase):
datasets = ["Boston", "Digits", "Cancer", "Sparse regression"]
common_param = {
'booster': ['gblinear'],
@@ -16,7 +15,7 @@ class TestGPULinear(unittest.TestCase):
'eta': [0.5],
'top_k': [10],
'tolerance': [1e-5],
'alpha': [.005, .1],
'alpha': [.1],
'lambda': [0.005],
'coordinate_selection': ['cyclic', 'random', 'greedy']}
@@ -26,6 +25,6 @@ class TestGPULinear(unittest.TestCase):
parameters['gpu_id'] = [0]
for param in test_linear.parameter_combinations(parameters):
results = test_linear.run_suite(
param, 150, self.datasets, scale_features=True)
param, 100, self.datasets, scale_features=True)
test_linear.assert_regression_result(results, 1e-2)
test_linear.assert_classification_result(results)