接下来, 我们从 Keras 中导入 CNN 层. 这些是卷积层, 将帮助我们有效地训练图片数据:
?
最后, 导入一下工具. 这将为我们之后的数据转换提供帮助:
?...可以像这样加载它:
?
看一下数据集的形状:
?
很好, 看来我们的训练集中有 60000 个样本, 每张图片都是 28*28 像素大小的....打印当前模型的输出进行确认:
?
然后, 我们可以像搭积木一样向模型中添加更多的层:
?
再次声明, 我们不会太深究理论的东西, 但有必要强调一下我们刚刚添加的 Dropout 层....到目前为止, 对于模型的参数, 我们已经添加了 2 个卷积层. 要完成模型的架构, 让我们添加一个完全连接的层和输出层:
?
对于 Dense 层, 第一个参数是输出的大小....只需要编译模型, 然后我们就可以训练它了. 编译模型时, 我们需要声明损失函数和优化器 (SGD, Adam 等等).
?
Keras 有各种各样的 损失函数和开箱即用的优化器.