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

Имеются данные матрицы X1,Y1. Задача: написать модель которую можно запустить, обучить на (X1,Y1), и получить предсказания из Х2, сравнить корреляцию предсказаний на основании X2 с Y2. Имеется архитектура нейросети и обучения, необходимо написать код для ее обучения на данных (Х1,Y1)  и получения предсказаний.Подробности ответившим.