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

logit cross entropy

Browse Source
This commit is contained in:
pevnak 2017-10-17 17:57:10 +01:00 committed by Mike J Innes
parent 6dff8ca8d3
commit 4aa7741ba9
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/layers/stateless.jl
View File

@ -6,3 +6,9 @@ crossentropy(ŷ::AbstractVecOrMat, y::AbstractVecOrMat) =
-sum(y .* log.(ŷ)) / size(y, 2) -sum(y .* log.(ŷ)) / size(y, 2)
@deprecate logloss(x, y) crossentropy(x, y) @deprecate logloss(x, y) crossentropy(x, y)
function logitcrossentropy(logŷ, y::AbstractMatrix, w)
logŷ = logŷ .-maximum(logŷ,1)
ypred = logŷ .- log.( sum( exp.( logŷ),1))
-sum(y .* w .* ypred)
end