Merge pull request #635 from rizar/fix633

Fix 633
mila-iqia · May 15, 2015 · 49f568a · 49f568a
2 parents e4d2cd0 + f563f9a
commit 49f568a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/blocks/bricks/cost.py b/blocks/bricks/cost.py
@@ -65,5 +65,8 @@ def apply(self, y, y_hat):
 class MisclassificationRate(Cost):
     @application(outputs=["error_rate"])
     def apply(self, y, y_hat):
-        return (tensor.sum(tensor.neq(y, y_hat.argmax(axis=1))) /
-                y.shape[0].astype(floatX))
+        # Here we have to cast both operands to floatX explicitly.
+        # Because int64 / float32 = float64 in Theano, unfortunately.
+        return (
+            tensor.sum(tensor.neq(y, y_hat.argmax(axis=1))).astype(floatX) /
+            y.shape[0].astype(floatX))