Код Junior SGD, momentum и Adam обновлено
SGD с momentum
Условие
Минимизируем f(w) = a·w². Напишите momentum(w0, a, lr, beta, steps): обновления v ← beta·v + g, w ← w - lr·v, старт v = 0. Верните [w, v] после steps шагов, округлённые до 4 знаков.
Примеры
momentum(1, 1, 0.1, 0.9, 1) // [0.8, 2]
momentum(1, 1, 0.1, 0.9, 2) // [0.46, 3.4]
momentum(1, 1, 0.1, 0, 3) // [0.512, 1.28]
Решите в тренажёре
Редактор с автотестами прямо в браузере. Можно решить спокойно, вслух или под таймером, как на собесе.