diff --git a/nolearn/lasagne/base.py b/nolearn/lasagne/base.py index e449a9b..77ec7bc 100644 --- a/nolearn/lasagne/base.py +++ b/nolearn/lasagne/base.py @@ -626,7 +626,7 @@ def _create_iter_funcs(self, layers, objective, update, output_type): predict_proba = get_output(output_layer, None, deterministic=True) if not self.regression: predict = predict_proba[0].argmax(axis=1) - accuracy = T.mean(T.eq(predict, y_batch)) + accuracy = T.mean(T.eq(predict, y_batch), dtype=theano.config.floatX) else: accuracy = loss_eval