Unverified Commit 477cbf37 authored by Nikita Titov's avatar Nikita Titov Committed by GitHub
Browse files

[python][tests] use default tolerance for dual GPU+CPU test (#3810)

* Update test_dual.py

* Update .vsts-ci.yml

* Update .vsts-ci.yml

* Update test_dual.py

* Update .vsts-ci.yml

* Update .vsts-ci.yml

* Update .vsts-ci.yml
parent 6bb6164e
...@@ -27,8 +27,9 @@ def test_cpu_and_gpu_work(): ...@@ -27,8 +27,9 @@ def test_cpu_and_gpu_work():
params_gpu = params_cpu.copy() params_gpu = params_cpu.copy()
params_gpu["device"] = "gpu" params_gpu["device"] = "gpu"
params_gpu["gpu_use_dp"] = True
gpu_bst = lgb.train(params_gpu, data, num_boost_round=10) gpu_bst = lgb.train(params_gpu, data, num_boost_round=10)
gpu_score = log_loss(y, gpu_bst.predict(X)) gpu_score = log_loss(y, gpu_bst.predict(X))
np.testing.assert_allclose(cpu_score, gpu_score, rtol=1e-4) np.testing.assert_allclose(cpu_score, gpu_score)
assert gpu_score < 0.25 assert gpu_score < 0.242
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment