Визуализация результатов нейронной сети на Keras

Всем привет, есть нейронная сеть на Keras для задачи регрессии, мне нужно построить график остатков и PredictionError график, однако Keras усредняет результаты на каждой эпохе обучения, есть какой либо способ получить предсказания для каждого сэмпла, а не усреднённое для всей эпохи? Находил гайд по построению этих график через sklearn и yellow brick, но метод там не очень простой, может быть есть вариант проще?

Спасибо за внимание.