search

SauceCat / PDPbox

python partial dependence plot toolbox
http://pdpbox.readthedocs.io/en/latest/
MIT License
840 stars 129 forks source link

Partial Dependence Plot kills the kernel w/ Xgboost #50

Closed jschular closed 3 years ago

jschular commented 5 years ago

Doesn't work w/ XGBClassifier. The kernel dies.

SauceCat commented 5 years ago

so it works for other classifiers rather than XGB?

BrunoGomesCoelho commented 5 years ago

Not sure if this is your error, but depending on the versions of your libraries, make sure to pass n_jobs = 1. From the pdpDocs (CTRL-F xgboost), it's a known problem with XGBoost