Mercurial > ift6266
view code_tutoriel/test.py @ 269:4533350d7361
Ajout d'une fonctionnalite pour pouvoir definir un range de fichiers d'entrainement de P07 utilises. Utile pour pre-entrainer et fine-tuner avec des donnees differentes
author | SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca> |
---|---|
date | Sat, 20 Mar 2010 10:19:11 -0400 |
parents | 4bc5eeec6394 |
children |
line wrap: on
line source
#import convolutional_mlp, dbn, logistic_cg, logistic_sgd, mlp, rbm, SdA_loops, SdA import convolutional_mlp, logistic_cg, logistic_sgd, mlp, SdA from nose.plugins.skip import SkipTest #TODO: dbn, rbm, SdA, SdA_loops, convolutional_mlp def test_logistic_sgd(): logistic_sgd.sgd_optimization_mnist(n_epochs=10) def test_logistic_cg(): logistic_cg.cg_optimization_mnist(n_epochs=10) def test_mlp(): mlp.test_mlp(n_epochs=5) def test_convolutional_mlp(): convolutional_mlp.evaluate_lenet5(n_epochs=5,nkerns=[5,5]) def test_dbn(): raise SkipTest('Implementation not finished') def test_rbm(): raise SkipTest('Implementation not finished') def test_SdA(): SdA.test_SdA(pretraining_epochs = 2, training_epochs = 3)