Question

我正在尝试运行一个用于对MNIST进行分类的简单网络：

network = models.Sequential()
network.add(layers.Dense(512, activation = 'relu', input_shape=(28 * 28,)))
network.add(layers.Dense(10, activation='softmax'))

当我尝试适应时：

network.fit(train_images, train_labels, epochs =5, batch_size = 128)

我收到此错误：

检查输入时出错：预期density_8_input具有2 尺寸，但数组的形状为（60000，28，28）

我在做什么？

Answer 1

您的模型期望每个输入样本的形状为methods:{ xhrSubmit(){ this.selected_to_show = true; //your code here } }（即(784,)）。但是，如错误所示，输入数组当前的形状为input_shape=(28 * 28,)。因此，您需要重塑它：

(num_samples, 28, 28)

检查输入时出错：预期density_8_input具有2维

1 个答案: