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

[backport] Fix inference with categorical feature. (#8591) (#8602) (#8638)

Browse Source 
* Fix inference with categorical feature. (#8591)

* Fix windows build on buildkite. (#8602)

* workaround.
This commit is contained in:
Jiaming Yuan
2023-01-06 01:17:49 +08:00
committed by GitHub
parent 1a834b2b85
commit 067b704e58
7 changed files with 79 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/python/test_with_sklearn.py
View File

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