wehub-resource-sync
26 lines
510 B
Python
26 lines
510 B
Python
import logging
|
|
|
|
import numpy as np
|
|
|
|
from mla.rbm import RBM
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
|
|
|
|
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))
|
|
rbm = RBM(n_hidden=10, max_epochs=200, batch_size=10, learning_rate=0.1)
|
|
rbm.fit(X)
|
|
print_curve(rbm)
|