From 89d07c07ec5c781de7525cd9f7a864ebe9978335 Mon Sep 17 00:00:00 2001 From: Adarsh Kumar <45385384+AdarshKumar712@users.noreply.github.com> Date: Mon, 2 Mar 2020 13:33:44 +0530 Subject: [PATCH] Added Loss functions to docs --- docs/src/models/layers.md | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/docs/src/models/layers.md b/docs/src/models/layers.md index 41e98f32..5522fe73 100644 --- a/docs/src/models/layers.md +++ b/docs/src/models/layers.md @@ -56,7 +56,10 @@ GroupNorm ## Cost Functions ```@docs +Flux.mae Flux.mse +Flux.msle +Flux.huber_loss Flux.crossentropy Flux.logitcrossentropy Flux.binarycrossentropy @@ -64,4 +67,7 @@ Flux.logitbinarycrossentropy Flux.kldivergence Flux.poisson Flux.hinge +Flux.squared_hinge +Flux.dice_coeff_loss +Flux.tversky_loss ```