plot_rewards.py

import matplotlib.pyplot as plt
import pandas as pd

df = pd.read_csv("saves/scores.csv")

plt.figure()
plt.plot(df['iter'], df['score'])
plt.savefig("saves/plot_rewards.png")