TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

loss function tweaks

Browse Source
This commit is contained in:
Mike J Innes 2017-10-17 17:30:11 +01:00
parent c764b74eba
commit 949fd9ba97
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/layers/stateless.jl
View File

@ -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)