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

[backport] Fix sklearn test that calls a removed field (#8579) (#8636)

Browse Source 
Co-authored-by: Rong Ou <rong.ou@gmail.com>
This commit is contained in:
Jiaming Yuan 2023-01-05 21:17:05 +08:00 committed by GitHub
parent 575fba651b
commit 83a078b7e5
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

@ -1030,7 +1030,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]))