aboutsummaryrefslogtreecommitdiff
path: root/model/dest_simple_mlp_tgtcls.py
diff options
context:
space:
mode:
authorAlex Auvolat <alex.auvolat@ens.fr>2015-05-22 10:00:15 -0400
committerAlex Auvolat <alex.auvolat@ens.fr>2015-05-22 10:00:15 -0400
commit39e549f05e568e4153381f025b3a0f256e9a7b7a (patch)
tree4e06cc45f359ad6a8cf4243fe7b42d01a23f7512 /model/dest_simple_mlp_tgtcls.py
parentdf50f103c1167f54a3ec04c1380fc95e4a023428 (diff)
downloadtaxi-39e549f05e568e4153381f025b3a0f256e9a7b7a.tar.gz
taxi-39e549f05e568e4153381f025b3a0f256e9a7b7a.zip
Make indexing faster by indexing only one column and querying a range
Diffstat (limited to 'model/dest_simple_mlp_tgtcls.py')
-rw-r--r--model/dest_simple_mlp_tgtcls.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/model/dest_simple_mlp_tgtcls.py b/model/dest_simple_mlp_tgtcls.py
index 2d65097..46fca2b 100644
--- a/model/dest_simple_mlp_tgtcls.py
+++ b/model/dest_simple_mlp_tgtcls.py
@@ -9,7 +9,7 @@ from model.mlp import FFMLP, Stream
class Model(FFMLP):
def __init__(self, config, **kwargs):
- super(Model, self, output_layer=Softmax).__init__(config, **kwargs)
+ super(Model, self).__init__(config, output_layer=Softmax, **kwargs)
self.classes = theano.shared(numpy.array(config.tgtcls, dtype=theano.config.floatX), name='classes')
@application(outputs=['destination'])