Python: машинное обучение. Натренировать модель.

Имеется архитектура нейросети и обучения предсказания стоимости объектов. Необходимо изучить код, предложить варианты улучшения модели и увеличения ее эффективности. Доработать код, протестировать, развернуть в системе (имеется интерфейс).

Модель построена на обучении данных на матрице X1,Y1 и тестируется на X2, Y2. 

Подробности ответившим.