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

Enable shap sparse test. (#6332)

Browse Source
This commit is contained in:
Jiaming Yuan 2020-11-01 20:59:27 +08:00 committed by GitHub
parent 5e1e972aea
commit 048acf81cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/python-gpu/test_gpu_prediction.py
View File

@ -212,10 +212,6 @@ class TestGPUPredict(unittest.TestCase):
tm.dataset_strategy, shap_parameter_strategy) tm.dataset_strategy, shap_parameter_strategy)
@settings(deadline=None, max_examples=20) @settings(deadline=None, max_examples=20)
def test_shap_interactions(self, num_rounds, dataset, param): def test_shap_interactions(self, num_rounds, dataset, param):
if dataset.name == 'sparse':
issue = 'https://github.com/dmlc/xgboost/issues/6074'
pytest.xfail(reason=f'GPU shap with sparse is flaky: {issue}')
param.update({"predictor": "gpu_predictor", "gpu_id": 0}) param.update({"predictor": "gpu_predictor", "gpu_id": 0})
param = dataset.set_params(param) param = dataset.set_params(param)
dmat = dataset.get_dmat() dmat = dataset.get_dmat()