sean_owen
Databricks Employee
Databricks Employee

Yes, TreeExplainer only works on the tree-based model itself. That's fine and the way to use it if you literally only want to explain the model, not the pipeline. If you want to explain anything else like a PIpeline or custom pyfunc model, you need to use KernelExplainer in SHAP (think it's just called Explainer now, yes). It's much slower but can operate on anything.