Блог

Когда я читал документы Flux, я увидел, что для нас определена куча различных функций потерь, которые мы можем использовать. Я понимаю, что потеря говорит нам о том, насколько мы далеки от целевого значения. Но где я на самом деле могу использовать функцию потерь в цикле обучения?

Если вы используете встроенную train!() функцию, вы можете определить свою функцию потерь и использовать ее во время тренировки следующим образом:

 loss(x, y) = Flux.Losses.mse(m(x), y)
ps = Flux.params(m)

Flux.train!(loss, ps, data, opt)
 

где Flux.Losses.mse используется встроенная функция среднеквадратичной ошибки для вычисления расстояния между m(x) и y . Вы можете прочитать больше о функциях потерь в потоке здесь: https://fluxml.ai/Flux.jl/stable/training/training/#Loss-Functions