Added Loss functions to docs
This commit is contained in:
parent
f9e31a020c
commit
89d07c07ec
@ -56,7 +56,10 @@ GroupNorm
|
|||||||
|
|
||||||
## Cost Functions
|
## Cost Functions
|
||||||
```@docs
|
```@docs
|
||||||
|
Flux.mae
|
||||||
Flux.mse
|
Flux.mse
|
||||||
|
Flux.msle
|
||||||
|
Flux.huber_loss
|
||||||
Flux.crossentropy
|
Flux.crossentropy
|
||||||
Flux.logitcrossentropy
|
Flux.logitcrossentropy
|
||||||
Flux.binarycrossentropy
|
Flux.binarycrossentropy
|
||||||
@ -64,4 +67,7 @@ Flux.logitbinarycrossentropy
|
|||||||
Flux.kldivergence
|
Flux.kldivergence
|
||||||
Flux.poisson
|
Flux.poisson
|
||||||
Flux.hinge
|
Flux.hinge
|
||||||
|
Flux.squared_hinge
|
||||||
|
Flux.dice_coeff_loss
|
||||||
|
Flux.tversky_loss
|
||||||
```
|
```
|
||||||
|
Loading…
Reference in New Issue
Block a user