aboutsummaryrefslogtreecommitdiff
path: root/model/rnn_lag_tgtcls.py
diff options
context:
space:
mode:
Diffstat (limited to 'model/rnn_lag_tgtcls.py')
-rw-r--r--model/rnn_lag_tgtcls.py31
1 files changed, 31 insertions, 0 deletions
diff --git a/model/rnn_lag_tgtcls.py b/model/rnn_lag_tgtcls.py
new file mode 100644
index 0000000..49d1f01
--- /dev/null
+++ b/model/rnn_lag_tgtcls.py
@@ -0,0 +1,31 @@
+import numpy
+import theano
+from theano import tensor
+from blocks.bricks.base import lazy
+from blocks.bricks import Softmax
+
+from model.rnn import RNN, Stream
+
+
+class Model(RNN):
+ @lazy()
+ def __init__(self, config, **kwargs):
+ super(Model, self).__init__(config, rec_input_len=4, output_dim=config.tgtcls.shape[0], **kwargs)
+ self.classes = theano.shared(numpy.array(config.tgtcls, dtype=theano.config.floatX), name='classes')
+ self.softmax = Softmax()
+ self.sequences.extend(['latitude_lag', 'longitude_lag'])
+ self.children.append(self.softmax)
+
+ def before_predict_all(self, kwargs):
+ super(Model, self).before_predict_all(kwargs)
+ kwargs['latitude_lag'] = tensor.extra_ops.repeat(kwargs['latitude'], 2, axis=0)
+ kwargs['longitude_lag'] = tensor.extra_ops.repeat(kwargs['longitude'], 2, axis=0)
+
+ def process_rto(self, rto):
+ return tensor.dot(self.softmax.apply(rto), self.classes)
+
+ def rec_input(self, latitude, longitude, latitude_lag, longitude_lag, **kwargs):
+ return (tensor.shape_padright(latitude),
+ tensor.shape_padright(longitude),
+ tensor.shape_padright(latitude_lag),
+ tensor.shape_padright(longitude_lag))