Keras - 依赖于历元的损失函数

时间:2017-03-24 09:26:15

标签: deep-learning keras autoencoder

我正在使用Keras框架,我想实现一个与历元相关的损失函数(即损失函数在每个时代都不相同)

你会怎么做?您可以添加一个示例,例如基于the keras VAE tutorial ?

感谢您的帮助

1 个答案:

答案 0 :(得分:2)

这可以通过重新编译网络来完成。保存的权重不会通过重新编译进行更改。所以本质上是这样的:

for epoch in range(nb_epoch):
     loss_function = loss_for_epoch(epoch)
     model.compile(optimizer, loss_function, metrics)
     model.fit(X, y, nb_epoch=1)