shap

app.py

@@ -112,9 +112,10 @@ with gr.Blocks() as demo:
         ## explainability plots
         rf_model = model.steps[-1][1]
         print("model: ", rf_model)
-        explainer = shap.Explainer(rf_model)
+        explainer = shap.TreeExplainer(rf_model, model_output="raw")
         shap_values = explainer(df)
-        print(shap_values.shape)  # should show (num_samples, 1)
+        print(shap_values.shape)  # should now show (1, 4)
+
         shap_waterfall_plot = shap.plots.waterfall(shap_values[0])
 
         ## save user's data in hopsworks