From b51d7aec4e5efdcb3fe2f3047e906fc109ba44f6 Mon Sep 17 00:00:00 2001
From: Antoine CARME <antoine.carme@outlook.com>
Date: Sat, 4 Feb 2023 20:04:43 +0100
Subject: [PATCH] Investigate Large Horizon Models #213. Added prediction
 Intervals movies for larger horizons (H=12, 24, 36, 48)

---
 tests/all_models_plots/model_esthetics_visualizer.py  |  2 +-
 .../visualize_air_passengers_models_H_24.py           | 11 +++++++++++
 .../visualize_air_passengers_models_H_36.py           | 11 +++++++++++
 .../visualize_air_passengers_models_H_48.py           | 11 +++++++++++
 tests/all_models_plots/visualize_ozone_models_H_24.py | 11 +++++++++++
 tests/all_models_plots/visualize_ozone_models_H_36.py | 11 +++++++++++
 tests/all_models_plots/visualize_ozone_models_H_48.py | 11 +++++++++++
 7 files changed, 67 insertions(+), 1 deletion(-)
 create mode 100644 tests/all_models_plots/visualize_air_passengers_models_H_24.py
 create mode 100644 tests/all_models_plots/visualize_air_passengers_models_H_36.py
 create mode 100644 tests/all_models_plots/visualize_air_passengers_models_H_48.py
 create mode 100644 tests/all_models_plots/visualize_ozone_models_H_24.py
 create mode 100644 tests/all_models_plots/visualize_ozone_models_H_36.py
 create mode 100644 tests/all_models_plots/visualize_ozone_models_H_48.py

diff --git a/tests/all_models_plots/model_esthetics_visualizer.py b/tests/all_models_plots/model_esthetics_visualizer.py
index 51c5ebc30..8e46cd25a 100644
--- a/tests/all_models_plots/model_esthetics_visualizer.py
+++ b/tests/all_models_plots/model_esthetics_visualizer.py
@@ -97,7 +97,7 @@ def generate_video(self, iDataset):
         images = sorted(images, key = lambda x : -x[1])
 
         import imageio as iio
-        writer = iio.get_writer(lSignal + "_models.mp4", fps=20)
+        writer = iio.get_writer(lSignal + "_models_H_" + str(self.mDataset.mHorizon)+ ".mp4", fps=20)
         for im in images:
             writer.append_data(im[0])
         writer.close()
diff --git a/tests/all_models_plots/visualize_air_passengers_models_H_24.py b/tests/all_models_plots/visualize_air_passengers_models_H_24.py
new file mode 100644
index 000000000..f818d9f32
--- /dev/null
+++ b/tests/all_models_plots/visualize_air_passengers_models_H_24.py
@@ -0,0 +1,11 @@
+
+
+import pyaf.Bench.TS_datasets as tsds
+import model_esthetics_visualizer as viz
+
+lDataset = tsds.load_airline_passengers()
+lDataset.mHorizon = 24
+
+lVisualizer = viz.cModelEstheticsVisualizer()
+
+lVisualizer.generate_video(lDataset)
diff --git a/tests/all_models_plots/visualize_air_passengers_models_H_36.py b/tests/all_models_plots/visualize_air_passengers_models_H_36.py
new file mode 100644
index 000000000..32f9f1a42
--- /dev/null
+++ b/tests/all_models_plots/visualize_air_passengers_models_H_36.py
@@ -0,0 +1,11 @@
+
+
+import pyaf.Bench.TS_datasets as tsds
+import model_esthetics_visualizer as viz
+
+lDataset = tsds.load_airline_passengers()
+lDataset.mHorizon = 36
+
+lVisualizer = viz.cModelEstheticsVisualizer()
+
+lVisualizer.generate_video(lDataset)
diff --git a/tests/all_models_plots/visualize_air_passengers_models_H_48.py b/tests/all_models_plots/visualize_air_passengers_models_H_48.py
new file mode 100644
index 000000000..79fd9c48b
--- /dev/null
+++ b/tests/all_models_plots/visualize_air_passengers_models_H_48.py
@@ -0,0 +1,11 @@
+
+
+import pyaf.Bench.TS_datasets as tsds
+import model_esthetics_visualizer as viz
+
+lDataset = tsds.load_airline_passengers()
+lDataset.mHorizon = 48
+
+lVisualizer = viz.cModelEstheticsVisualizer()
+
+lVisualizer.generate_video(lDataset)
diff --git a/tests/all_models_plots/visualize_ozone_models_H_24.py b/tests/all_models_plots/visualize_ozone_models_H_24.py
new file mode 100644
index 000000000..7ec221728
--- /dev/null
+++ b/tests/all_models_plots/visualize_ozone_models_H_24.py
@@ -0,0 +1,11 @@
+
+
+import pyaf.Bench.TS_datasets as tsds
+import model_esthetics_visualizer as viz
+
+lDataset = tsds.load_ozone()
+lDataset.mHorizon = 24
+
+lVisualizer = viz.cModelEstheticsVisualizer()
+
+lVisualizer.generate_video(lDataset)
diff --git a/tests/all_models_plots/visualize_ozone_models_H_36.py b/tests/all_models_plots/visualize_ozone_models_H_36.py
new file mode 100644
index 000000000..f518f4aa4
--- /dev/null
+++ b/tests/all_models_plots/visualize_ozone_models_H_36.py
@@ -0,0 +1,11 @@
+
+
+import pyaf.Bench.TS_datasets as tsds
+import model_esthetics_visualizer as viz
+
+lDataset = tsds.load_ozone()
+lDataset.mHorizon = 36
+
+lVisualizer = viz.cModelEstheticsVisualizer()
+
+lVisualizer.generate_video(lDataset)
diff --git a/tests/all_models_plots/visualize_ozone_models_H_48.py b/tests/all_models_plots/visualize_ozone_models_H_48.py
new file mode 100644
index 000000000..824840d51
--- /dev/null
+++ b/tests/all_models_plots/visualize_ozone_models_H_48.py
@@ -0,0 +1,11 @@
+
+
+import pyaf.Bench.TS_datasets as tsds
+import model_esthetics_visualizer as viz
+
+lDataset = tsds.load_ozone()
+lDataset.mHorizon = 48
+
+lVisualizer = viz.cModelEstheticsVisualizer()
+
+lVisualizer.generate_video(lDataset)