Keras 深度学习 - 模型编译

编译是使用一个名为compile的单一方法调用来执行的。

model.compile(loss='categorical_crossentropy', metrics=['accuracy'], optimizer='adam')

compile方法需要几个参数。loss 参数指定为类型'categorical_crossentropy'。metrics 参数设置为'accuracy'，最后我们使用adam优化器来训练网络。此阶段的输出如下所示：

现在，我们准备将数据馈送到我们的网络中。

加载数据

如前所述，我们将使用 Keras 提供的mnist数据集。当我们将数据加载到我们的系统中时，我们将将其拆分为训练数据和测试数据。数据通过调用load_data方法加载，如下所示：

(X_train, y_train), (X_test, y_test) = mnist.load_data()

此阶段的输出如下所示：

现在，我们将学习加载数据集的结构。

提供给我们的数据是大小为 28 x 28 像素的图形图像，每个图像包含 0 到 9 之间的一个数字。我们将在控制台上显示前十张图像。执行此操作的代码如下：

# printing first 10 images
for i in range(10):

plot.subplot(3,5,i+1)
plot.tight_layout()
plot.imshow(X_train[i], cmap='gray', interpolation='none')
plot.title("Digit: {}".format(y_train[i]))
plot.xticks([])
plot.yticks([])

在 10 次计数的迭代循环中，我们在每次迭代中创建一个子图，并在其中显示来自X_train向量的图像。我们从相应的y_train向量为每个图像设置标题。请注意，y_train向量包含X_train向量中相应图像的实际值。我们通过调用两个方法xticks和yticks并使用空参数来删除 x 和 y 轴标记。运行代码时，您将看到以下输出：

接下来，我们将准备数据以将其馈送到我们的网络中。