我尝试在VGG16网络上训练CIFAR10和MNIST数据集。在第一次尝试中,我遇到了一个错误,提示输入input_2(标签)的形状必须为(None,2,2,10)。这个2x2x10的数组结构包含了什么信息呢?因为我期望input_2的形状是(None, 10)(我的两个数据集中都有10个类别)。我尝试将标签的维度从(None,10)扩展到(None,2,2,10)。但我确定这不是正确的方法,因为我得到了非常低的准确率(大约0.09)。
(我使用的是Keras,Python3.5)
回答:
低准确率是由层的问题引起的。我只是修改了我的网络,并获得了0.7496的准确率。