训练首个神经网络：基本分类

Posted 2021-02-10 jtay

在Fashion MNIST数据集上使用tf.keras完成对服饰图像分类的神经网络模型。

1，导入数据集

数据大小：70000张 灰度图像， 10个类别 28*28的像素

==> 60000张图像 用于训练， 10000张 用来评估

2，探索数据

3，预处理数据

灰度值 0-255 缩小到 0-1

4，构建模型

1）配置层

扁平化像素： 28*28 ==> 1* 784 转换成1维数组 

第一个Dense层具有 128个节点  即 784*128

第二个Dense层具有 10个节点 即 128 * 10

softmax层 ==> 返回具有10个概率得分的数组

2）编译模型

* 损失函数，衡量模型在训练期间的准确率

* 优化器，根据模型看到的数据及其损失函数更新模型的方式

* 指标，用于监控训练和测试步骤

5，训练模型

1） 将训练数据递送到模型中 model.fit()

2）模型学习将图像和标签相关联

在模型训练期间，系统会显示损失和准确率指标。该模型在训练数据上的准确率达到 0.88（即 88%）

3）使用模型对测试集进行预测

6，评估准确率

测试集上的准确率为0.87 < 0.88

结果表明，模型在测试数据集上的准确率略低于在训练数据集上的准确率。

==> 过拟合

7，做出预测

模型上线，对未标注的图像进行预测

代码见链接：

https://github.com/shuqingjinse/NLP_homework_2/blob/master/simple_tensorflow.py