在 Keras 中训练期间更改每个批次的输入数据

Question

我有一个 Keras model 和一个给定的输入数据集。

在每次批量训练之前，我需要根据 model 的预测和在 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，它允许您将 function 设置为

• on_epoch_begin：在每个纪元开始时调用。
• on_epoch_end：在每个 epoch 结束时调用。
• on_batch_begin：在每批开始时调用。
• on_batch_end：在每批结束时调用。
• on_train_begin：在 model 训练开始时调用。 on_train_end：在 model 训练结束时调用。