def test_transform():
X = Xdigits[:100].astype(np.float32)
rbm1 = BernoulliRBM(X.shape[1], n_hidden=16, batch_size=5,
n_iter=5, random_state=42)
rbm1.fit(X)
Xt1 = rbm1.transform(X)
Xt2 = rbm1._mean_hiddens(X)
assert_array_equal(Xt1, Xt2)
def test_transform():
X = Xdigits[:100].astype(np.float32)
rbm1 = BernoulliRBM(X.shape[1],
n_hidden=16,
batch_size=5,
n_iter=5,
random_state=42)
rbm1.fit(X)
Xt1 = rbm1.transform(X)
Xt2 = rbm1._mean_hiddens(X)
assert_array_equal(Xt1, Xt2)
