def setUp(self): self.n_samples = 10 self.layer = DARN( n_X=16, n_Y=8, ) self.layer.setup()
class TestDARN(RWSLayerTest, unittest.TestCase): def setUp(self): self.n_samples = 10 self.layer = DARN( n_X=16, n_Y=8, ) self.layer.setup()
dataset = MNIST(fname="data/mnist_salakhutdinov.pkl.gz", which_set='salakhutdinov_train', n_datapoints=59000) smallset = MNIST(fname="data/mnist_salakhutdinov.pkl.gz", which_set='salakhutdinov_valid', n_datapoints=100) valiset = MNIST(fname="data/mnist_salakhutdinov.pkl.gz", which_set='salakhutdinov_valid', n_datapoints=1000) testset = MNIST(fname="data/mnist_salakhutdinov.pkl.gz", which_set='test', n_datapoints=10000) p_layers = [ DARN( n_X=n_vis, n_Y=200, ), DARNTop(n_X=200, ), ] q_layers = [ CNADE( n_Y=n_vis, n_X=200, n_hid=200, ), ] model = LayerStack( p_layers=p_layers, q_layers=q_layers,