|
1 | | -#import convolutional_mlp, dbn, logistic_cg, logistic_sgd, mlp, rbm, SdA_loops, SdA |
2 | 1 | import convolutional_mlp, logistic_cg, logistic_sgd, mlp, SdA, dA |
3 | 2 | from nose.plugins.skip import SkipTest |
4 | | -import time,sys |
5 | | -#TODO: dbn, rbm, SdA, SdA_loops, convolutional_mlp |
| 3 | +import theano |
| 4 | +#TODO: rbm, DBN, deep |
| 5 | + |
6 | 6 | def test_logistic_sgd(): |
7 | | - t0=time.time() |
8 | 7 | logistic_sgd.sgd_optimization_mnist(n_epochs=10) |
9 | | - print >> sys.stderr, "test_logistic_sgd took %.3fs expected 15.2s in our buildbot"%(time.time()-t0) |
10 | 8 | def test_logistic_cg(): |
11 | | - t0=time.time() |
12 | 9 | logistic_cg.cg_optimization_mnist(n_epochs=10) |
13 | | - print >> sys.stderr, "test_logistic_cg took %.3fs expected 14s in our buildbot"%(time.time()-t0) |
14 | 10 | def test_mlp(): |
15 | | - t0=time.time() |
16 | 11 | mlp.test_mlp(n_epochs=5) |
17 | | - print >> sys.stderr, "test_mlp took %.3fs expected 118s in our buildbot"%(time.time()-t0) |
18 | 12 | def test_convolutional_mlp(): |
19 | | - t0=time.time() |
20 | 13 | convolutional_mlp.evaluate_lenet5(n_epochs=5,nkerns=[5,5]) |
21 | | - print >> sys.stderr, "test_convolutional_mlp took %.3fs expected 168s in our buildbot"%(time.time()-t0) |
22 | 14 | def test_dbn(): |
23 | 15 | raise SkipTest('Implementation not finished') |
24 | 16 | def test_rbm(): |
25 | 17 | raise SkipTest('Implementation not finished') |
26 | 18 | def test_dA(): |
27 | | - t0=time.time() |
28 | | - dA.test_dA(training_epochs = 3) |
29 | | - print >> sys.stderr, "test_dA took %.3fs expected Xs in our buildbot"%(time.time()-t0) |
| 19 | + dA.test_dA(training_epochs = 2) |
30 | 20 | def test_SdA(): |
31 | | - t0=time.time() |
32 | | - SdA.test_SdA(pretraining_epochs = 2, training_epochs = 3) |
33 | | - print >> sys.stderr, "test_SdA took %.3fs expected 971s in our buildbot"%(time.time()-t0) |
| 21 | + SdA.test_SdA(pretraining_epochs = 1, training_epochs = 2) |
0 commit comments