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