diff --git a/utils.py b/utils.py index bbd2bf0063f6a00f72a481aaa0478718eaca71c8..b45c9bccf2030e7dca3209464b0caf9fad6ce678 100644 --- a/utils.py +++ b/utils.py @@ -58,7 +58,7 @@ def plot2D_assistance(save_path, n_episodes, *y): plt.legend() # Custom X axis plt.xticks(r, x, fontweight='bold') - plt.ylabel("performance") + plt.ylabel("assistance") plt.savefig(save_path) plt.show() @@ -79,6 +79,6 @@ def plot2D_feedback(save_path, n_episodes, *y): plt.legend() # Custom X axis plt.xticks(r, x, fontweight='bold') - plt.ylabel("performance") + plt.ylabel("feedback") plt.savefig(save_path) plt.show() \ No newline at end of file