plot.py 文件源码

def plot_return(agent, returns, data=None):
    ''' Plot return over time. '''
    plt.plot(returns, agent.colour, label=agent.legend)
    interval = 5000
    if data != None:
        for i in range(returns.size/interval):
            plt.errorbar(1+i*interval, returns[i*interval],
                yerr=t.sem(data[:, i*interval]), fmt=agent.colour)
    plt.axis([0, returns.size, 0.0, 0.8])
    plt.xlabel('Episodes')
    plt.title('Average Return')
    plt.ylabel('Average Return')