Add predictor to skl constructor. (#7000)

This commit is contained in:
Jiaming Yuan
2021-05-29 04:52:56 +08:00
committed by GitHub
parent 55b823b27d
commit 816b789bf0
3 changed files with 12 additions and 2 deletions

View File

@@ -374,6 +374,7 @@ class XGBModel(XGBModelBase):
importance_type: str = "gain",
gpu_id: Optional[int] = None,
validate_parameters: Optional[bool] = None,
predictor: Optional[str] = None,
**kwargs: Any
) -> None:
if not SKLEARN_INSTALLED:
@@ -409,6 +410,7 @@ class XGBModel(XGBModelBase):
self.importance_type = importance_type
self.gpu_id = gpu_id
self.validate_parameters = validate_parameters
self.predictor = predictor
def _more_tags(self) -> Dict[str, bool]:
'''Tags used for scikit-learn data validation.'''