科研小trick,早停技巧,用过的都说好~,爸爸妈妈再也不怕我过拟合啦

Early stopping

  • 在我们训练神经网络的时候,若出现过拟合的情况下,如果我们将验证集的验证误差和训练集的训练误差画图出来,将呈现下图所示的情况
  • image-20211215210959787
  • 可以看出,在过拟合的时候,验证误差随着迭代次数的增加一般呈现出先下降,然后再上升的一个趋势
  • 通过画出这么一个图后,我们就可以找到验证误差上升到那个点的迭代次数,然后只取迭代这么多次的参数来防止过拟合,这就是early stopping的主要思想
  • 有了这个思想呢,在你训练模型的时候,你就可以在代码中进行设置,若训练过程中x次(我一般设置为3次)在验证集测试的时候若你的评价指标性能一直没有提高,则直接保存模型,停止训练,从而保存当前参数设置的最佳模型,不然的话很有可能你的模型按照你最初设置的epoch一直训练下去然后就过拟合了
  • 16
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值