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

Update document for sklearn model IO. (#6809)

Browse Source 
* Update the use of JSON.
* Remove unnecessary type cast.
This commit is contained in:
Jiaming Yuan
2021-04-01 15:52:36 +08:00
committed by GitHub
parent 905fdd3e08
commit a5c852660b
3 changed files with 19 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/python/test_with_sklearn.py
View File

@@ -804,10 +804,14 @@ def save_load_model(model_path):
for train_index, test_index in kf.split(X, y):
xgb_model = xgb.XGBClassifier(use_label_encoder=False).fit(X[train_index], y[train_index])
xgb_model.save_model(model_path)
xgb_model = xgb.XGBClassifier(use_label_encoder=False)
xgb_model = xgb.XGBClassifier()
xgb_model.load_model(model_path)
assert xgb_model.use_label_encoder is False
assert isinstance(xgb_model.classes_, np.ndarray)
assert isinstance(xgb_model._Booster, xgb.Booster)
preds = xgb_model.predict(X[test_index])
labels = y[test_index]
err = sum(1 for i in range(len(preds))