Fix/ShapExplainer-summary_plot-Horizon-does-not-include-output_chunk_…

…shift (#2647) * Add output_chunk_shift to summary_plot title * Update CHANGELOG.md * Update darts/explainability/shap_explainer.py Co-authored-by: Dennis Bader <[email protected]> --------- Co-authored-by: Dennis Bader <[email protected]>
unit8co · Jan 21, 2025 · c52ec83 · c52ec83
1 parent 0a52490
commit c52ec83
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -17,6 +17,7 @@ but cannot always guarantee backwards compatibility. Changes that may **break co
 
 **Fixed**
 - Fixed a bug when performing optimized historical forecasts with `stride=1` using a `RegressionModel` with `output_chunk_shift>=1` and `output_chunk_length=1`, where the forecast time index was not properly shifted. [#2634](https://github.com/unit8co/darts/pull/2634) by [Mattias De Charleroy](https://github.com/MattiasDC).
+- Fixed the `ShapExplainer` `summary_plot` title where Horizon does not include `output_chunk_shift`. [#2647](https://github.com/unit8co/darts/pull/2647) by [He Weilin](https://github.com/cnhwl).
 
 **Dependencies**
 

diff --git a/darts/explainability/shap_explainer.py b/darts/explainability/shap_explainer.py
@@ -420,7 +420,11 @@ def summary_plot(
 
         for t in target_components:
             for h in horizons:
-                plt.title("Target: `{}` - Horizon: {}".format(t, "t+" + str(h)))
+                plt.title(
+                    "Target: `{}` - Horizon: {}".format(
+                        t, "t+" + str(h + self.model.output_chunk_shift)
+                    )
+                )
                 shap.summary_plot(
                     shaps_[h][t],
                     foreground_X_sampled,