experiment.py 文件源码

def plot_episode_reward():
    pylab.clf()
    sns.set_context("poster")
    pylab.plot(0, 0)
    episodes = [0]
    scores = [0]
    for n in xrange(len(csv_episode)):
        params = csv_episode[n]
        episodes.append(params[0])
        scores.append(params[1])
    pylab.plot(episodes, scores, sns.xkcd_rgb["windows blue"], lw=2)
    pylab.xlabel("episodes")
    pylab.ylabel("score")
    pylab.savefig("%s/episode_reward.png" % args.plot_dir)