diff --git a/docs/src/models/regularisation.md b/docs/src/models/regularisation.md index d4325a53..70d06348 100644 --- a/docs/src/models/regularisation.md +++ b/docs/src/models/regularisation.md @@ -7,6 +7,7 @@ add the result to the overall loss. For example, say we have a simple regression. ```julia +using Flux: crossentropy m = Dense(10, 5) loss(x, y) = crossentropy(softmax(m(x)), y) ```