Код Junior Градиентный спуск и выпуклость обновлено
Спуск по вытянутой параболе
Условие
f(x, y) = x² + k·y². Напишите gd2(x, y, k, lr, steps): steps шагов градиентного спуска, верните [x, y, f] в конце, округлённые до 4 знаков. Сравните, что происходит при k = 1 и k = 25 с одним и тем же lr.
Примеры
gd2(1, 1, 1, 0.1, 1) // [0.8, 0.8, 1.28]
gd2(1, 1, 1, 0.1, 50) // [0, 0, 0]
gd2(1, 1, 25, 0.1, 5) // [0.3277, -1024, 26214400.1074]
Решите в тренажёре
Редактор с автотестами прямо в браузере. Можно решить спокойно, вслух или под таймером, как на собесе.
Предыдущий вопрос
Как подобрать learning rate
Следующий вопрос
Обучите линейную регрессию градиентным спуском