From 9013799ebca1c426c3c3e9019eb71018b253b025 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=89tienne=20Simon?= <esimon@esimon.eu>
Date: Tue, 21 Jul 2015 21:08:35 -0400
Subject: Add bidirectional

---
 model/bidirectional_tgtcls.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)
 create mode 100644 model/bidirectional_tgtcls.py

(limited to 'model/bidirectional_tgtcls.py')

diff --git a/model/bidirectional_tgtcls.py b/model/bidirectional_tgtcls.py
new file mode 100644
index 0000000..36120f7
--- /dev/null
+++ b/model/bidirectional_tgtcls.py
@@ -0,0 +1,19 @@
+import numpy
+import theano
+from theano import tensor
+from blocks.bricks.base import lazy
+from blocks.bricks import Softmax
+
+from model.bidirectional import BidiRNN, Stream
+
+
+class Model(BidiRNN):
+    @lazy()
+    def __init__(self, config, **kwargs):
+        super(Model, self).__init__(config, 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.children.append(self.softmax)
+
+    def process_outputs(self, outputs):
+        return tensor.dot(self.softmax.apply(outputs), self.classes)
-- 
cgit v1.2.3