JuliaML / LossFunctions.jl

Julia package of loss functions for machine learning.
https://juliaml.github.io/LossFunctions.jl/stable
Other
148 stars 34 forks source link

Derivative of cross-entropy loss #135

Closed deepaksuresh closed 4 years ago

deepaksuresh commented 4 years ago

Isn't the derivative supposed to be output*(1-output) rather than what's defined here?

>>deriv(CrossEntropyLoss(), 0,0.1)
1.1111111111111112

Shouldn't the derivative in this case be 0.09?