diff options
| author | Étienne Simon <esimon@esimon.eu> | 2015-06-21 17:26:03 -0400 |
|---|---|---|
| committer | Étienne Simon <esimon@esimon.eu> | 2015-06-21 17:26:03 -0400 |
| commit | ad5a03c6f60e5b2d543326bf8917b48e5b390b82 (patch) | |
| tree | 0da344a44d0bdb1b2e59d36db3f1a81ed2e1e5fa /model/memory_network.py | |
| parent | 33f1543d7fb87d47467a296b9408b84c5440b186 (diff) | |
| download | taxi-ad5a03c6f60e5b2d543326bf8917b48e5b390b82.tar.gz taxi-ad5a03c6f60e5b2d543326bf8917b48e5b390b82.zip | |
Forgot a transpose in memory network
Diffstat (limited to 'model/memory_network.py')
| -rw-r--r-- | model/memory_network.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model/memory_network.py b/model/memory_network.py index c17f9b3..92e83e2 100644 --- 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( |