Skip to content

Commit 44d2912

Browse files
cleanup
1 parent f5386f1 commit 44d2912

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

docs/src/models/losses.md

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,14 +10,15 @@ can be imported with
1010
using 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 ``.
1514
In Flux's convention, the order of the arguments is the following:
1615

1716
```julia
1817
loss(ŷ, y)
1918
```
2019

20+
They are commonly passed as arrays of size `num_target_features x num_examples_in_batch`.
21+
2122
Most loss functions in Flux have an optional argument `agg`, denoting the type of aggregation performed over the
2223
batch:
2324

0 commit comments

Comments
 (0)