From 877703b0dc411ba60c9febd021528e39db594037 Mon Sep 17 00:00:00 2001 From: Wenjian Huang Date: Mon, 14 Dec 2015 09:40:29 +0800 Subject: [PATCH] Update bench_mnist.py --- examples/bench_mnist.py | 12 +----------- 1 file changed, 1 insertion(+), 11 deletions(-) diff --git a/examples/bench_mnist.py b/examples/bench_mnist.py index 5f15117..8a0314b 100644 --- a/examples/bench_mnist.py +++ b/examples/bench_mnist.py @@ -6,7 +6,7 @@ import numpy as np if len(sys.argv) == 1: - print("ERROR: Please specify implementation to benchmark, 'sknn' 'dbn' or 'lasagne'.") + print("ERROR: Please specify implementation to benchmark, 'sknn' or 'lasagne'.") sys.exit(-1) np.set_printoptions(precision=4) @@ -27,16 +27,6 @@ classifiers = [] -if 'dbn' in sys.argv: - from nolearn.dbn import DBN - clf = DBN( - [X_train.shape[1], 300, 10], - learn_rates=0.3, - learn_rate_decays=0.9, - epochs=10, - verbose=1) - classifiers.append(('nolearn.dbn', clf)) - if 'sknn' in sys.argv: from sknn.mlp import Classifier, Layer, Convolution