reshapes kernel shap output

this deals with e.g. PLSRegression
oegedijk · May 9, 2023 · 3f4a9a0 · 3f4a9a0
1 parent b2d3f07
commit 3f4a9a0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/explainerdashboard/explainers.py b/explainerdashboard/explainers.py
@@ -1107,7 +1107,8 @@ def shap_explainer(self):
 
                 def model_predict(data_asarray):
                     data_asframe = pd.DataFrame(data_asarray, columns=self.columns)
-                    return self.model.predict(data_asframe)
+                    preds = self.model.predict(data_asframe)
+                    return preds.reshape(len(preds))
 
                 self._shap_explainer = shap.KernelExplainer(
                     model_predict,