Commit 6a283888 authored by Daniel Lukats's avatar Daniel Lukats

minor fix to evaluation script

parent 071e16a9
......@@ -102,7 +102,6 @@ def create_plot(experiment: Experiment,
:param plot:
:return:
"""
# colors = ['#307B3B', '#CAA023', '#254796', '#D01B88']
colors = ['#307B3B', '#CAA023', '#254796', '#F1009A']
setup()
for i, run in enumerate(experiment.runs):
......@@ -110,7 +109,7 @@ def create_plot(experiment: Experiment,
plt.plot(comparison[0], comparison[1], color='black', label='Trendline')
plt.title(f'{experiment.game}')
plt.xlabel('Time step')
plt.ylabel('Score')
plt.ylabel('Episode score')
make_legend(colors, trend=False)
plt.tight_layout(pad=1.02)
......@@ -122,7 +121,7 @@ def create_plot(experiment: Experiment,
if default_trend is not None:
plt.plot(default_trend[0], default_trend[1], label='Trendline (reference)', color=colors[3])
make_legend(colors)
plt.tight_layout(pad=1.08)
plt.tight_layout(pad=1.02)
plt.savefig(path + '_reference')
plt.clf()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment