Monitor and Recorder are Wrappers.

2021-11-24 17:39:26 +01:00
parent 59484f49c9
commit b0d6c2e1ef
10 changed files with 241 additions and 350 deletions
--- a/plotting/plotting.py
+++ b/plotting/plotting.py
@ -39,7 +39,9 @@ def prepare_plt(df, hue, style, hue_order):
    plt.close('all')
    sns.set(rc={'text.usetex': False}, style='whitegrid')
    lineplot = sns.lineplot(data=df, x='Episode', y='Score', hue=hue, style=style,
-                            ci=95, palette=PALETTE, hue_order=hue_order)
+                            ci=95, palette=PALETTE, hue_order=hue_order, )
+    plt.legend(bbox_to_anchor=(1.02, 1), loc='upper left', borderaxespad=0)
+    plt.tight_layout()
    # lineplot.set_title(f'{sorted(list(df["Measurement"].unique()))}')
    return lineplot