def testInit(useTheano=False):
n_groups = 16
n_components = 24 * 24
window_size = 5
learning_rate = 0.1
batch_size = 10
n_iter = 1000
verbose = False
r = convRBM(n_groups=n_groups,
n_components=n_components,
window_size=window_size,
learning_rate=learning_rate,
batch_size=batch_size,
n_iter=n_iter,
verbose=verbose,
use_theano=useTheano)
return r
def testRunMnist():
n_groups = 16
n_components = 24 * 24
window_size = 5
learning_rate = 0.1
batch_size = 50
n_iter = 20
r = convRBM(n_groups=n_groups,
n_components=n_components,
window_size=window_size,
learning_rate=learning_rate,
batch_size=batch_size,
n_iter=n_iter,
verbose=False)
digits = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
images, labels = mn.load_mnist('training', digits,
'/Users/jiajunshen/Dropbox/Research/data/',
False, slice(0, 6000, 5), True, False)
return r
