From e98b0764d1bc15fc89fc8cffe6c7c56c62398e6f Mon Sep 17 00:00:00 2001
From: bkn <nico.beck@iis.fraunhofer.de>
Date: Fri, 4 Oct 2024 11:52:38 +0200
Subject: [PATCH] Improve resolution of uncertainty heatmap

---
 examples/case_study/visualization.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/examples/case_study/visualization.py b/examples/case_study/visualization.py
index e66a91f..55c63ff 100644
--- a/examples/case_study/visualization.py
+++ b/examples/case_study/visualization.py
@@ -33,8 +33,8 @@
     forecast_horizon,
     split_date=train_test_split_period,
     data_type="test",
-    target=target,
 )
+fredmd_test.target = target
 
 # %% Visualization of the ECNN ensemble Heatmap
 
@@ -68,8 +68,8 @@
     forecasts.squeeze(2),
     start_point=start_point,
     sigma=0.25,
-    num_interp=30,
-    window_height=200,
+    num_interp=100,
+    window_height=1000,
     xlabel="Forecast step",
     save_at=Path(__file__).parent / f"heatmap_{target}.pdf",
     title="Heatmap ensemble forecast",