神经网络训练与超参数调优指南
1. 监测训练与验证损失比
在训练过程中,监测验证损失与训练损失的比率有助于检测过拟合。可以通过自定义回调函数来实现这一功能,示例代码如下:
class PrintValTrainRatioCallback(keras.callbacks.Callback):
def on_epoch_end(self, epoch, logs):
print("\nval/train: {:.2f}".format(logs["val_loss"] / logs["loss"]))
此外,还可以实现 on_train_begin() 、 on_train_end() 、 on_epoch_begin() 、 on_epoch_end() 、 on_batch_begin() 和 on_batch_end() 等方法。在评估和预测阶段,也可以使用回调函数,比如在评估时实现 on_test_begin() 、 on_test_end() 、 on_test_batch_begin() 或 on_test_batch_end() ,在预测时实现 on_predict_begin() 、 on_predict_end()
超级会员免费看
订阅专栏 解锁全文
619

被折叠的 条评论
为什么被折叠?



