shap

app.py

@@ -111,9 +111,12 @@ with gr.Blocks() as demo:
 
         ## explainability plots
         rf_model = model.steps[-1][1]
-        print("model: ", rf_model)
-        explainer = shap.TreeExplainer(rf_model, model_output="raw")
+
+        explainer = shap.TreeExplainer(
+            rf_model, model_output="raw", feature_perturbation="tree_path_dependent"
+        )
         shap_values = explainer(df)
+
         print(shap_values.shape)  # should now show (1, 4)
 
         shap_waterfall_plot = shap.plots.waterfall(shap_values[0])