From 9ce0439943325e7efc6df3655b5f855772788aaf Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Thu, 24 Aug 2017 11:40:51 +0100 Subject: [PATCH] better mse --- src/layers/stateless.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/layers/stateless.jl b/src/layers/stateless.jl index 52db7c7c..e1fe49b1 100644 --- a/src/layers/stateless.jl +++ b/src/layers/stateless.jl @@ -1,6 +1,6 @@ # Cost functions -mse(ŷ, y) = sum((ŷ .- y).^2)/2 +mse(ŷ, y) = sum((ŷ .- y).^2)/length(y) # back!(::typeof(mse), Δ, ŷ, y) = Δ .* (ŷ .- y) logloss(ŷ, y) = -sum(y .* log.(ŷ))