steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-11-02 13:37:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

fixed prepare_plot and added LaTeX support

Browse Source
This commit is contained in:
romue
2021-06-02 10:31:49 +02:00
parent 38a3ef7687
commit c798226b26
2 changed files with 14 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
environments/logging/plotting.py
View File

@@ -18,9 +18,7 @@ PALETTE = 10 * (
)
def plot(filepath, ext='png', **kwargs):
plt.rcParams.update(kwargs)
def plot(filepath, ext='png'):
plt.tight_layout()
figure = plt.gcf()
figure.savefig(str(filepath), format=ext)
@@ -29,26 +27,14 @@ def plot(filepath, ext='png', **kwargs):
def prepare_plot(filepath, results_df, ext='png'):
sns.set_theme(palette=PALETTE, 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
}
sns.lineplot(data=results_df, x='Episode', y='Score', hue='Measurement', ci='sd')
results_df.Measurement = results_df.Measurement.str.replace('_', '-')
try:
plot(filepath, ext=ext, **tex_fonts)
sns.set(rc={'text.usetex': True}, style='whitegrid')
sns.lineplot(data=results_df, x='Episode', y='Score', hue='Measurement', ci='sd', palette=PALETTE)
plot(filepath, ext=ext) # plot raises errors not lineplot!
except (FileNotFoundError, RuntimeError):
tex_fonts['text.usetex'] = False
plot(filepath, ext=ext, **tex_fonts)
print('Struggling to plot Figure using LaTeX - going back to normal.')
plt.close('all')
sns.set(rc={'text.usetex': False}, style='whitegrid')
sns.lineplot(data=results_df, x='Episode', y='Score', hue='Measurement', ci='sd', palette=PALETTE)
plot(filepath, ext=ext)