search

micahmelling / auto-shap

MIT License
11 stars 1 forks source link

Does it support deep learning models? #3

Open KuangHao95 opened 1 year ago

KuangHao95 commented 1 year ago

Tried to run this with a pytorch model, which can be done in Shap's DeepExplainer, but gives error:

AttributeError: 'my_model_name' object has no attribute 'predict_proba'

Want to know whether it supports deep explainer and how to do it?

Thank you.

micahmelling commented 1 year ago

My apologies for the long delay on this response. Presently, auto-shap does not support deep learning models, but this is something I can look at for an upcoming release. I will follow up on this issue with a resolution.