训练的时候为什么要shuffle?训练和验证又有什么区别呢?

训练时shuffle数据集能避免模型记住数据顺序导致的过拟合,增加梯度优化方向的选择性,促进模型收敛到全局最优点。训练集用于模型学习和误差减少,通常需打乱数据;验证集则用于超参数调整,评估模型性能。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

训练的时候为什么要shuffle?

当 dataloader 的参数 shuffle 设置为 true 时,表明每次训练的数据集的顺序是一致的,模型会记住这种顺序,从而产生过拟合现象。
这里有一个例子可以用来讲解:
在这里插入图片描述
图中的 A 点,每次权值更新的方向是一致的,则可能会沿着红色箭头走到局部最优点,但如果打乱数据集后,每次权值更新方向是随机的,则可能会沿着白色箭头走到全局最优点,更有利于模型的收敛。
因此,固定的数据集顺序,严重限制了梯度优化方向的可选择性,导致收敛点选择空间严重变少,容易导致过拟合。

训练和验证有什么区别?

模型是从训练集中学习经验,从而不断减少训练误差,为了加快模型的收敛,因此要打乱数据;
验证集用于调整超参数,用于在训练过程中检验模型的状态与收敛情况,根据几组模型验证集上的表现决定哪组超参数拥有最好的性能。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值