class TestDARNTop(RWSTopLayerTest, unittest.TestCase): def setUp(self): self.n_samples = 10 self.layer = DARNTop( n_X=8, ) self.layer.setup()
def setUp(self): self.n_samples = 10 self.layer = DARNTop(n_X=8, ) self.layer.setup()
class TestDARNTop(RWSTopLayerTest, unittest.TestCase): def setUp(self): self.n_samples = 10 self.layer = DARNTop(n_X=8, ) self.layer.setup()
def setUp(self): self.n_samples = 10 self.layer = DARNTop( n_X=8, ) self.layer.setup()
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, )