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