Function moving_average

examples/rbm.py:13–16

(a, n=25)

    from matplotlib import pyplot as plt

    def moving_average(a, n=25):
        ret = np.cumsum(a, dtype=float)
        ret[n:] = ret[n:] - ret[:-n]
        return ret[n - 1 :] / n

    plt.plot(moving_average(rbm.errors))
    plt.show()