Added framework for saving charts in plot_improvement.py

plot_improvement.py

@@ -16,6 +16,7 @@ def calc_stats(filepath):
 
 # parser = argparse.ArgumentParser()
 # parser.add_argument("-f", "--filepath", required=True, help="Specify the file path to the agent.", type=str)
+# parser.add_argument("-s", "--save", help="Specify whether to save the chart.", action="store_const", const=True)
 # args = parser.parse_args()
 
 # Get the file paths and store in list.
@@ -44,7 +45,8 @@ plt.ylabel("score")
 plt.bar(runs, means)
 plt.bar(runs, stds)
 plt.legend(["Mean evaluation score", "Standard deviation"])
-plt.title("Average Evaluation Score and Standard Deviation\nAdjusted for Outliers\nAgent: dqn_v2")
+plt.title("Average Evaluation Score and Standard Deviation\nAdjusted for Outliers   Agent: dqn_v2")
 plt.show()
+# plt.savefig("charts/fig1")