Esempio n. 1
0
 def test_gpu_coordinate(self):
     parameters = self.common_param.copy()
     parameters['n_gpus'] = [1]
     for param in test_linear.parameter_combinations(parameters):
         results = test_linear.run_suite(
             param, 150, self.datasets, scale_features=True)
         test_linear.assert_regression_result(results, 1e-2)
         test_linear.assert_classification_result(results)
Esempio n. 2
0
 def test_gpu_coordinate(self):
     parameters = self.common_param.copy()
     parameters['gpu_id'] = [0]
     for param in test_linear.parameter_combinations(parameters):
         results = test_linear.run_suite(
             param, 150, self.datasets, scale_features=True)
         test_linear.assert_regression_result(results, 1e-2)
         test_linear.assert_classification_result(results)
Esempio n. 3
0
 def test_gpu_coordinate(self):
     tm._skip_if_no_sklearn()
     variable_param = {'booster': ['gblinear'], 'updater': ['coord_descent'], 'eta': [0.5],
                       'top_k': [10], 'tolerance': [1e-5], 'nthread': [2], 'alpha': [.005, .1], 'lambda': [0.005],
                       'coordinate_selection': ['cyclic', 'random', 'greedy'], 'n_gpus': [-1]}
     for param in test_linear.parameter_combinations(variable_param):
         results = test_linear.run_suite(param, 200, None, scale_features=True)
         test_linear.assert_regression_result(results, 1e-2)
         test_linear.assert_classification_result(results)