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

Fix sklearn test that calls a removed field (#8579)

Browse Source
This commit is contained in:
Rong Ou 2022-12-09 13:06:44 -08:00 committed by GitHub
parent deb3edf562
commit 42e6fbb0db
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/python/test_with_sklearn.py
View File

@ -1025,7 +1025,7 @@ def test_pandas_input():
clf_isotonic = CalibratedClassifierCV(model, cv="prefit", method="isotonic") clf_isotonic = CalibratedClassifierCV(model, cv="prefit", method="isotonic")
clf_isotonic.fit(train, target) clf_isotonic.fit(train, target)
assert isinstance( assert isinstance(
clf_isotonic.calibrated_classifiers_[0].base_estimator, xgb.XGBClassifier clf_isotonic.calibrated_classifiers_[0].estimator, xgb.XGBClassifier
) )
np.testing.assert_allclose(np.array(clf_isotonic.classes_), np.array([0, 1])) np.testing.assert_allclose(np.array(clf_isotonic.classes_), np.array([0, 1]))