Investigate Large Horizon Models #213. Added prediction Intervals mov…

…ies for larger horizons (H=12, 24, 36, 48)
antoinecarme · Feb 4, 2023 · b51d7ae · b51d7ae
1 parent 1177454
commit b51d7ae
Show file tree

Hide file tree

Showing 7 changed files with 67 additions and 1 deletion.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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)