commit
422b31c8f7
@ -7,6 +7,7 @@ add the result to the overall loss.
|
|||||||
For example, say we have a simple regression.
|
For example, say we have a simple regression.
|
||||||
|
|
||||||
```julia
|
```julia
|
||||||
|
using Flux: crossentropy
|
||||||
m = Dense(10, 5)
|
m = Dense(10, 5)
|
||||||
loss(x, y) = crossentropy(softmax(m(x)), y)
|
loss(x, y) = crossentropy(softmax(m(x)), y)
|
||||||
```
|
```
|
||||||
|
Loading…
Reference in New Issue
Block a user