Keras 教程目录
下面是创建你的第一个卷积神经网络 (CNN) 的步骤:
配置环境
安装 Keras
导入库和模块
从 MNIST 导入图片数据
预处理输入数据
预处理类标签
定义模型架构
编译模型...可以再次打印 X_train 的维度, 进行确认:
?
输入数据最后的预处理步骤是, 将数据类型转换成 float32, 并且进行标准化, 使值落在 [0, 1] 区间内.
?...问题依旧. y_train 和 y_test 数据并没有被分成 10 个不同的类标签, 而是被表示成一个类型值的单一数组.
对此, 我们可以进行简单的修正:
?
再看一眼:
?
好多了!...第九步: 用训练数据进行模型拟合
要拟合模型, 我们需要做的就是声明训练的批次大小以及训练次数, 然后传入训练数据.
?
简单吗?...你也可以使用各种回调函数来设置提前结束的规则, 保存模型权重, 或记录每次训练的历史.
第十步: 用测试数据评估模型
最后, 可以用测试数据对模型进行评估:
?
恭喜!