Я тоже обучил модель на игрушечном примере. По номеру строки и столбца картинки, c единственным скрытым слоем из 5 нейронов, предсказывается выход по трем цветовым каналам для аппроксимации картинки. Тут показана обученная модель с графиками функций на ребрах и ее предикт.