如何在 Flux.jl 中提前停止

Question

使用 Flux.jl，有没有一种方法可以通过实施某种提前停止功能来避免过度拟合？

Answer 1

Flux.jl 提供了一个内置的 Flux.early_stopping 函数，可以按如下方式使用：

julia> loss = let l = 0
         () -> l += 1
       end; # pseudo loss function that returns increasing values

julia> es = Flux.early_stopping(loss, 3);

julia> Flux.@epochs 30 begin
         es() && break
       end

[ Info: Epoch 1

您可以在此处阅读更多相关信息：https://fluxml.ai/Flux.jl/stable/utilities/#Flux.early_stopping 了解有关如何在 Flux 中实现提前停止的更多详细信息。