Fix filtering callable objects in skl xgb param. (#6466)

Co-authored-by: Hyunsu Cho <chohyu01@cs.washington.edu>
This commit is contained in:
Jiaming Yuan
2020-12-05 17:20:36 +08:00
committed by Hyunsu Cho
parent 2b3e301543
commit c39f6b25f0
2 changed files with 16 additions and 1 deletions

View File

@@ -398,7 +398,7 @@ class XGBModel(XGBModelBase):
'importance_type', 'kwargs', 'missing', 'n_estimators', 'use_label_encoder'}
filtered = dict()
for k, v in params.items():
if k not in wrapper_specific:
if k not in wrapper_specific and not callable(v):
filtered[k] = v
return filtered