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

Fixed issue 3605. (#3628)

Browse Source 
* Fixed issue 3605.

- https://github.com/dmlc/xgboost/issues/3605

* Fixed the bug in a better way.

* Added a test to catch the bug.

* Fixed linter errors.
This commit is contained in:
Andy Adinets
2018-08-28 19:50:52 +02:00
committed by Philip Hyunsu Cho
parent 78bea0d204
commit 58d783df16
3 changed files with 18 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/python-gpu/test_gpu_updaters.py
View File

@@ -14,7 +14,7 @@ def assert_gpu_results(cpu_results, gpu_results):
assert np.allclose(cpu_res["eval"][-1], gpu_res["eval"][-1], 1e-2, 1e-2)
datasets = ["Boston", "Cancer", "Digits", "Sparse regression",
"Sparse regression with weights"]
"Sparse regression with weights", "Small weights regression"]
class TestGPU(unittest.TestCase):
def test_gpu_exact(self):