Removed unecessary comments in plot_improvement.py

plot_improvement.py

@@ -1,10 +1,9 @@
-import argparse
+# import argparse
 import numpy as np
 import os
 from matplotlib import pyplot as plt
 
 def calc_stats(filepath):
-    # load the numpy file
     data = np.load(filepath)["results"]
     # sort the arrays and delete the first and last elements
     data = np.sort(data, axis=1)
@@ -19,11 +18,6 @@ def calc_stats(filepath):
 # 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.
-# For each file path, I want to calculate the mean reward. This would be the mean reward for the training run over all evaluations.
-# For each file path, append the mean reward to an averages list
-# Plot the averages!
-
 filepaths = []
 for d in os.listdir("agents/"):
     if "dqn_v2" in d:
@@ -40,8 +34,8 @@ for path in filepaths:
 runs = []
 for i in range(len(filepaths)):
     runs.append(i + 1)
-plt.xlabel("training runs")
-plt.ylabel("score")
+plt.xlabel("Training Run")
+plt.ylabel("Score")
 plt.bar(runs, means)
 plt.bar(runs, stds)
 plt.legend(["Mean evaluation score", "Standard deviation"])