debug

ivysoftware · Mar 19, 2016 · e3bd91c · e3bd91c
1 parent 80d8a50
commit e3bd91c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/algorithms/dialog/GRU_2NN_recons.py b/algorithms/dialog/GRU_2NN_recons.py
@@ -167,13 +167,13 @@ def getDeployFunction(self, cr):
 
         recons_h_error_L = T.tanh(media_h) - T.addbroadcast(self.last_decoder_hidden, 0)
         recons_h_error_L = T.sqr(recons_h_error_L).sum(axis=1) 
-        recons_h_error_L = recons_h_error_L/ self.options['hidden_dim']
+        recons_h_error_L = recons_h_error_L / self.options['hidden_dim']
         error = -T.log(pred_word_probability) + recons_h_error_L
-
+        score = -error
         pred_word = T.argmin(error)
 
         deploy_model = theano.function(inputs=[self.answer, self.answer_mask, self.last_hidden_state],
-                                       outputs=[pred_word, error],
+                                       outputs=[pred_word, score],
                                        allow_input_downcast=True)
         print "Compiled."
         def dm(sentence):