Implement GPU accelerated coordinate descent algorithm (#3178)

* Implement GPU accelerated coordinate descent algorithm. 

* Exclude external memory tests for GPU
This commit is contained in:
Rory Mitchell
2018-04-20 14:56:35 +12:00
committed by GitHub
parent ccf80703ef
commit a185ddfe03
12 changed files with 473 additions and 63 deletions

View File

@@ -0,0 +1,14 @@
import sys
sys.path.append('tests/python/')
import test_linear
import testing as tm
import unittest
class TestGPULinear(unittest.TestCase):
def test_gpu_coordinate(self):
tm._skip_if_no_sklearn()
variable_param = {'alpha': [.005, .1], 'lambda': [0.005],
'coordinate_selection': ['cyclic', 'random', 'greedy'], 'n_gpus': [-1, 1]}
test_linear.assert_updater_accuracy('gpu_coord_descent', variable_param)