julia - 如何在 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 中实现提前停止的更多详细信息。