taxi

commit ad5a03c6f60e5b2d543326bf8917b48e5b390b82
parent 33f1543d7fb87d47467a296b9408b84c5440b186
Author: Étienne Simon <esimon@esimon.eu>
Date:   Sun, 21 Jun 2015 17:26:03 -0400

Forgot a transpose in memory network

Diffstat:
M
model/memory_network.py
 | 
2
+-

1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/model/memory_network.py b/model/memory_network.py
@@ -52,7 +52,7 @@ class Model(Initializable):
         candidate_inputs = tensor.concatenate(candidate_extremities + candidate_embeddings, axis=1)
         candidate_representation = self.candidate_encoder.apply(candidate_inputs)
 
-        similarity_score = tensor.dot(prefix_representation, candidate_representation)
+        similarity_score = tensor.dot(prefix_representation, candidate_representation.T)
         similarity = self.softmax.apply(similarity_score)
 
         candidate_destination = tensor.concatenate(