From 5dce8df6780e72c6cd916ae0808e75d168cf95f6 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Mon, 1 May 2017 13:40:11 +0100
Subject: [PATCH] fix accuracy for batches

---
 src/training.jl | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/training.jl b/src/training.jl
index b0daf211..c8deb238 100644
--- a/src/training.jl
+++ b/src/training.jl
@@ -2,12 +2,14 @@ tobatch(xs::Batch) = rawbatch(xs)
 tobatch(xs) = tobatch(batchone(xs))
 
 function accuracy(m, data)
+  n = 0
   correct = 0
   for (x, y) in data
     x, y = tobatch.((x, y))
+    n += size(x, 1)
     correct += sum(onecold(m(x)) .== onecold(y))
   end
-  return correct/length(data)
+  return correct/n
 end
 
 function train!(m, train, test = [];