diff --git a/src/cost.jl b/src/cost.jl index b08675d5..a75b546c 100644 --- a/src/cost.jl +++ b/src/cost.jl @@ -3,3 +3,7 @@ export mse, mse! mse(ŷ, y) = sumabs2(ŷ .- y)/2 back!(::typeof(mse), Δ, ŷ, y) = Δ*(ŷ .- y) + +logloss(ŷ, y) = -sum(y .* log.(ŷ)) + +back!(::typeof(logloss), Δ, ŷ, y) = 0 .- Δ .* y ./ ŷ