Function print_curve

examples/rbm.py:10–19

(rbm)

def print_curve(rbm):
    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()


X = np.random.uniform(0, 1, (1500, 10))