You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The pathfinding task depends on the order of the answers but the code above makes the accuracy measurement independent from the order of pred[0], pred[1], pred[2] and a_s[0], a_s[1], a_s[2].
The text was updated successfully, but these errors were encountered:
From babi_story/module.py:
final_pred = tf.one_hot(tf.argmax(pred[0], axis=1), depth=self.answer_vocab_size) * answer_bool[0] + tf.one_hot(tf.argmax(pred[1], axis=1), depth=self.answer_vocab_size) * answer_bool[1] + tf.one_hot(tf.argmax(pred[2], axis=1), depth=self.answer_vocab_size) * answer_bool[2]
final_answer = a_s[0] * answer_bool[0] + a_s[1] * answer_bool[1] + a_s[2] * answer_bool[2]
The pathfinding task depends on the order of the answers but the code above makes the accuracy measurement independent from the order of pred[0], pred[1], pred[2] and a_s[0], a_s[1], a_s[2].
The text was updated successfully, but these errors were encountered: