Я пытаюсь создать нейронную сеть в Керасе (python) следующим образом:
ввод -> .... (плотные слои) ... -> вывод
Кроме того, я хочу, чтобы в моей модели была обучаемая переменная pi, которая будет оптимизирована для начального значения, которое я даю.
Точнее, функция потерь = MSE (вход, выход) + f (вход, pi) где f доступно в хорошем, аналити c форме. В качестве конкретного примера:
вход = n чисел x_i, без промежуточных слоев, n выводит y_i с функцией потерь = $$ \ sum_i (x_i-y_i) ^ 2 + pi ^ 2 * \ sum_i x_i ^ 2 $ $
Здесь pi следует оптимизировать с помощью (S) GD до значения 0.
Я попытался реализовать пользовательский слой, содержащий обучаемую переменную pi, или передать его как отдельный тензорный вход, но мне пока не удалось заставить код работать. Спасибо!