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
10
10
using Flux. Losses: logitcrossentropy
11
11
```
12
12
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 ` ŷ ` .
15
14
In Flux's convention, the order of the arguments is the following:
16
15
17
16
``` julia
18
17
loss (ŷ, y)
19
18
```
20
19
20
+ They are commonly passed as arrays of size ` num_target_features x num_examples_in_batch ` .
21
+
21
22
Most loss functions in Flux have an optional argument ` agg ` , denoting the type of aggregation performed over the
22
23
batch:
23
24
You can’t perform that action at this time.
0 commit comments