File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -10,14 +10,15 @@ can be imported with
1010using Flux. Losses: logitcrossentropy
1111```
1212
13- Loss functions for supervised learning typically expect as inputs a true target ` y ` and a prediction ` ŷ ` ,
14- typically passed as arrays of size ` num_target_features x num_examples_in_batch ` .
13+ Loss functions for supervised learning typically expect as inputs a true target ` y ` and a prediction ` ŷ ` .
1514In Flux's convention, the order of the arguments is the following:
1615
1716``` julia
1817loss (ŷ, y)
1918```
2019
20+ They are commonly passed as arrays of size ` num_target_features x num_examples_in_batch ` .
21+
2122Most loss functions in Flux have an optional argument ` agg ` , denoting the type of aggregation performed over the
2223batch:
2324
You can’t perform that action at this time.
0 commit comments