CNN模型训练结果不错,但并未很好的表现在分类上

你好。CNN对样本不平衡问题很敏感。在你的数据中,0,1两类肯定占到90%以上,使用的损失函数又是一般的交叉熵(alexnet),在训练过程中模型很快就学会忽略第三类,而只输出0,1两类。建议:对少数类进行过采样。且至少保证最少的类的样本数为最多类的一半。


    推荐阅读