Изменение входных данных каждой партии во время тренировки в Керасе

Question

У меня есть модель Keras и заданный набор входных данных.

Мне нужно изменить входные данные на основе прогноза из модели и процедуры линейного программирования, реализованной в Python, перед каждым пакетным обучением.

Кажется, что я мог бы сделать это перед каждой эпохой, используя цикл for, такой:

for each epoch:
  train model
  compute prediction based on input data
  new input data = LP(input data, prediction)
  break if epoch=50

Мой вопрос заключается в том, как сделать это перед каждой партией.

Answer 1

Возможно, вы захотите проверить Keras Обратные вызовы и особенно LambdaCallback. Он позволяет вам установить функцию для

• on_epoch_begin: вызываться в начале каждой эпохи.
• on_epoch_end: вызывается в конце каждой эпохи.
• on_batch_begin: вызывается в начале каждого пакета.
• on_batch_end: вызывается в конце каждого пакета.
• on_train_begin: вызывается в начале обучения модели. on_train_end: вызывается в конце обучения модели.