Mercurial > pylearn

comparison pylearn/algorithms/tests/test_mcRBM.py @ 1508:b28e8730c948

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
fix test.
author Frederic Bastien <nouiz@nouiz.org>
date Mon, 12 Sep 2011 11:45:56 -0400
parents 2a6a6f16416c
children b709f6b53b17
comparison
equal deleted inserted replaced
1507:2a6a6f16416c 1508:b28e8730c948
13 13
14 14
15 def _default_rbm_alloc(n_I, n_K=256, n_J=100): 15 def _default_rbm_alloc(n_I, n_K=256, n_J=100):
16 return mcRBM.alloc(n_I, n_K, n_J) 16 return mcRBM.alloc(n_I, n_K, n_J)
17 17
18 def _default_trainer_alloc(rbm, train_batch, batchsize, l1_penalty, l1_penalty_start): 18 def _default_trainer_alloc(rbm, train_batch, batchsize, initial_lr_per_example,
19 l1_penalty, l1_penalty_start, persistent_chains):
19 return mcRBMTrainer.alloc(rbm, train_batch, batchsize, l1_penalty=l1_penalty, 20 return mcRBMTrainer.alloc(rbm, train_batch, batchsize, l1_penalty=l1_penalty,
20 l1_penalty_start=l1_penalty_start,persistent_chains=persistent_chains) 21 l1_penalty_start=l1_penalty_start,persistent_chains=persistent_chains)
21 22
22 23
23 def test_reproduce_ranzato_hinton_2010(dataset='MAR', as_unittest=True, n_train_iters=5000, 24 def test_reproduce_ranzato_hinton_2010(dataset='MAR', as_unittest=True, n_train_iters=5000,