fix the T.neq not giving int8 type issue #29

uoguelph-mlrg · Mar 28, 2017 · db33b0f · db33b0f
1 parent 23440c6
commit db33b0f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/layers.py b/lib/layers.py
@@ -309,6 +309,6 @@ def errors_top_x(self, y, num_top=5):
             # represents a mistake in prediction
             y_pred_top_x = T.argsort(self.p_y_given_x, axis=1)[:, -num_top:]
             y_top_x = y.reshape((y.shape[0], 1)).repeat(num_top, axis=1)
-            return T.mean(T.min(T.neq(y_pred_top_x, y_top_x), axis=1))
+            return T.mean(T.min(T.neq(y_pred_top_x, y_top_x).astype('int8'), axis=1))
         else:
             raise NotImplementedError()