plotting

environments/logging/plotting.py

@@ -0,0 +1,40 @@
+import seaborn as sns
+
+from matplotlib import pyplot as plt
+
+
+def plot(filepath, ext='png', tag='monitor', **kwargs):
+    plt.rcParams.update(kwargs)
+
+    plt.tight_layout()
+    figure = plt.gcf()
+    plt.show()
+    figure.savefig(str(filepath.parent / f'{filepath.stem}_{tag}_measures.{ext}'), format=ext)
+
+
+def prepare_plot(filepath, results_df, ext='png', tag=''):
+    # %%
+
+    _ = sns.lineplot(data=results_df)
+
+    # %%
+    sns.set_theme(palette='husl', style='whitegrid')
+    font_size = 16
+    tex_fonts = {
+        # Use LaTeX to write all text
+        "text.usetex": False,
+        "font.family": "serif",
+        # Use 10pt font in plots, to match 10pt font in document
+        "axes.labelsize": font_size,
+        "font.size": font_size,
+        # Make the legend/label fonts a little smaller
+        "legend.fontsize": font_size - 2,
+        "xtick.labelsize": font_size - 2,
+        "ytick.labelsize": font_size - 2
+    }
+
+    try:
+        plot(filepath, ext=ext, tag=tag, **tex_fonts)
+    except (FileNotFoundError, RuntimeError):
+        tex_fonts['text.usetex'] = False
+        plot(filepath, ext=ext, tag=tag, **tex_fonts)