fix shap interaction values iteration

oegedijk · Mar 11, 2024 · 5813376 · 5813376
1 parent 17a3ee6
commit 5813376
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/explainerdashboard/explainers.py b/explainerdashboard/explainers.py
@@ -3105,7 +3105,7 @@ def shap_interaction_values(self, pos_label=None):
                 ):
                     self._shap_interaction_values = [
                         self._shap_interaction_values[:, :, :, i]
-                        for i in range(self._shap_interaction_values.shape)
+                        for i in range(self._shap_interaction_values.shape[3])
                     ]
                 assert len(self._shap_interaction_values) == len(self.labels), (
                     f"len(self.label)={len(self.labels)}, but "