我想在每个周期后更改输入,并在最后绘制学习曲线。
为了更改输入,我有一个可以按以下方式使用的函数
for _ in range(num_epochs): x, y = generate_data() history = model.fit(x, y, epochs=1, batch_size=64)
但我无法捕获模型的完整历史记录。我如何访问它并绘制学习曲线?
回答:
您可以使用字典来存储history
。
history_dict = dict()for i in range(num_epochs): x, y = generate_data() history_dict['epoch_%i' % i] = model.fit(x, y, epochs=1, batch_size=64)
它会生成类似这样的结果:
Out[4]: {'epoch_0': <your history 1>, 'epoch_1': <your history 2>, 'epoch_2': <your history 3>,