third_codes/eland
1
0
Fork 0
mirror of https://github.com/elastic/eland.git synced 2025-07-11 00:02:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add test case for pseudohubererror for XGBoost

Browse Source
This commit is contained in:
P. Sai Vinay 2021-08-07 02:29:48 +05:30 committed by GitHub
parent d3f8d7b8f6
commit 8f84a315be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/ml/test_ml_model_pytest.py
View File

@ -336,7 +336,13 @@ class TestMLModel:
@pytest.mark.parametrize("compress_model_definition", [True, False])
@pytest.mark.parametrize(
"objective",
["reg:squarederror", "reg:squaredlogerror", "reg:linear", "reg:logistic"],
[
"reg:squarederror",
"reg:squaredlogerror",
"reg:linear",
"reg:logistic",
"reg:pseudohubererror",
],
)
@pytest.mark.parametrize("booster", ["gbtree", "dart"])
def test_xgb_regressor(self, compress_model_definition, objective, booster):