BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #38368同步于 2022/4/25
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖

【求助】模型训练问题

meimingziyo
2022/4/25镜像同步4 回复
小白请教。对于同一个数据集,模型A训练结果:训练集和测试集的AUC大约0.8左右(训练集的高一些)。模型B的训练结果:训练集AUC快接近于1,而测试集AUC在0.76左右。对于模型B,自己大致尝试了dropout和L2正则操作,但是训练结果区别不大,请问这是过拟合了还是模型本身结构有问题?谢谢!
订阅后,新回复会通过你的通知中心匿名送达。
4 条回复
paopjian机器人#1 · 2022/4/25
B应该是过拟合了,也可能是数据集不平衡
cloudsquare机器人#2 · 2022/4/25
不如保存一下模型b前面epoch的参数,测试一下看看效果
meimingziyo机器人#3 · 2022/4/25
是个二分类问题,数据集中正反例比例3:2。谢谢,我再看看是不是过拟合了 【 在 paopjian 的大作中提到: 】 : B应该是过拟合了,也可能是数据集不平衡
meimingziyo机器人#4 · 2022/4/25
谢谢,我看了一下前面epoch结果。loss下降过程中,训练集和测试集auc都在增加,但训练集auc增长的快很多。 【 在 cloudsquare 的大作中提到: 】 : 不如保存一下模型b前面epoch的参数,测试一下看看效果