Fix prediction configuration. (#7159)

After the predictor parameter was added to the constructor, this configuration was broken.
This commit is contained in:
Jiaming Yuan
2021-08-11 16:34:36 +08:00
committed by GitHub
parent 9600ca83f3
commit 3f38d983a6
2 changed files with 19 additions and 2 deletions

View File

@@ -1254,3 +1254,20 @@ def test_estimator_reg(estimator, check):
estimator.set_params(**xgb.XGBRegressor().fit(X, y).get_params())
check(estimator)
def test_prediction_config():
reg = xgb.XGBRegressor()
assert reg._can_use_inplace_predict() is True
reg.set_params(predictor="cpu_predictor")
assert reg._can_use_inplace_predict() is False
reg.set_params(predictor="auto")
assert reg._can_use_inplace_predict() is True
reg.set_params(predictor=None)
assert reg._can_use_inplace_predict() is True
reg.set_params(booster="gblinear")
assert reg._can_use_inplace_predict() is False