From 949fd9ba97238dc93f89868363a51d78f05b0c6e Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Tue, 17 Oct 2017 17:30:11 +0100 Subject: [PATCH] loss function tweaks --- src/layers/stateless.jl | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/layers/stateless.jl b/src/layers/stateless.jl index 12523afe..76ade52c 100644 --- a/src/layers/stateless.jl +++ b/src/layers/stateless.jl @@ -1,7 +1,6 @@ # Cost functions mse(ŷ, y) = sum((ŷ .- y).^2)/length(y) -# back!(::typeof(mse), Δ, ŷ, y) = Δ .* (ŷ .- y) -logloss(ŷ, y) = -sum(y .* log.(ŷ)) / size(y, 2) -# back!(::typeof(logloss), Δ, ŷ, y) = 0 .- Δ .* y ./ ŷ +logloss(ŷ::AbstractVecOrMat, y::AbstractVecOrMat) = + -sum(y .* log.(ŷ)) / size(y, 2)