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

logitcrossentropy tweaks

Browse Source
This commit is contained in:
Mike J Innes 2017-10-17 17:58:32 +01:00
parent 4aa7741ba9
commit 23674b2555
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/layers/stateless.jl
View File

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