scores = pd.read_csv(args.scores, delimiter="\t")
for col in ["mean", "median"]:
plt.plot(scores["steps"], scores[col], label=col)
plt.title("A3C Breakout")
plt.xlabel("steps")
plt.ylabel("score")
plt.legend(loc="best")
// plt.show()
After Change
scores = pd.read_csv(args.scores, delimiter="\t")
for col in ["mean", "median"]:
plt.plot(scores["steps"], scores[col], label=col)
if args.title is not None:
plt.title(args.title)
plt.xlabel("steps")
plt.ylabel("score")
plt.legend(loc="best")
fig_fname = args.scores + ".png"