【发布时间】:2020-04-21 11:27:10
【问题描述】:
我在 pytorch 中的神经网络训练变得非常奇怪。
我正在训练一个已知数据集,该数据集分为训练和验证。 我在训练期间对数据进行洗牌,并在飞行中进行数据扩充。
我有这些结果:
我有以下图表要显示:
如何解释验证损失增加和验证准确度增加?
验证集和训练集之间的准确度差异为何如此之大? 90% 和 40%?
更新:
我平衡了数据集。 是二分类。现在它有来自第 1 类的 1700 个示例,来自第 2 类的 1200 个示例。总共有 600 个用于验证,2300 个用于训练。 我仍然看到类似的行为:
**可能是因为我冻结了部分网络的权重吗?
**可能是因为像 lr 这样的超参数吗?
【问题讨论】:
-
二分类还是更多类?
-
二进制分类
-
我猜是不平衡的,对吧?
-
它们是平衡的(3000 vs 1200)
-
很难说没有数据。实验为王。对于初学者,我会增加 val 集的大小,以“确定”指标是可靠的,并且不会因为样本量小而产生虚假。
标签: machine-learning neural-network pytorch training-data loss