crflynn / skgrf

scikit-learn compatible Python bindings for grf (generalized random forests) C++ random forest library
https://skgrf.readthedocs.io/en/stable/
GNU General Public License v3.0
30 stars 6 forks source link

explainer.shap_values crashes kernel #90

Open kochbj opened 2 years ago

kochbj commented 2 years ago

Hi,

Amazing package! I was wondering if you have any insight into why calling explainer.shap_values consistently crashes my python kernel. I don't think it's a memory issue.

Thanks for your time; really appreciate it.

-Bernie

crflynn commented 2 years ago

Hard to say without having more details.

I took a look at the shap tests here and noticed that when calling shap_values, check_additivity=False is passed. IIRC this was necessary otherwise it would crash, and I wasn't able to figure out why. Try including this argument.