From e268a61fdc9c3d487038a96be7cbdeab90c0e1df Mon Sep 17 00:00:00 2001
From: Alexandra Udaltsova <43303448+AUdaltsova@users.noreply.github.com>
Date: Mon, 5 Aug 2024 16:14:37 +0100
Subject: [PATCH] Update analysis.py

---
 experiments/analysis.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/experiments/analysis.py b/experiments/analysis.py
index 9d7d35f9..33fc718d 100644
--- a/experiments/analysis.py
+++ b/experiments/analysis.py
@@ -92,10 +92,10 @@ def main(runs: list[str], run_names: list[str]) -> None:
     plt.figure()
     for idx, df in enumerate(dfs):
         plt.plot(column_timesteps, df, label=f"{run_names[idx]}, epoch: {epoch_num[idx]}")
-    plt.legend(fontsize=18)
-    plt.xlabel("Timestep (minutes)", fontsize=18)
-    plt.ylabel("MAE %", fontsize=18)
-    plt.title("MAE % for each timestep", fontsize=24)
+    plt.legend()
+    plt.xlabel("Timestep (minutes)")
+    plt.ylabel("MAE %")
+    plt.title("MAE % for each timestep")
     plt.savefig("mae_per_timestep.png")
     plt.show()
 
@@ -103,10 +103,10 @@ def main(runs: list[str], run_names: list[str]) -> None:
     plt.figure()
     for idx, run_name in enumerate(run_names):
         plt.plot(groups_df[run_name], label=f"{run_name}, epoch: {epoch_num[idx]}")
-    plt.legend(fontsize=18)
-    plt.xlabel("Timestep (minutes)", fontsize=18)
-    plt.ylabel("MAE %", fontsize=18)
-    plt.title("MAE % for each timestep", fontsize=24)
+    plt.legend()
+    plt.xlabel("Timestep (minutes)")
+    plt.ylabel("MAE %")
+    plt.title("MAE % for each timestep")
     plt.savefig("mae_per_timestep.png")
     plt.show()