1.加载fashion_mnist数据集
- fashion_mnist是一个替代mnist手写数字集的图像数据集,包含60000个示例的训练集和10000个示例的测试集。每个示例都是一个28x28灰度图像,分为10个类别(’T-shirt’, ‘Trouser’, ‘Pullover’, ‘Dress’, ‘Coat’, ‘Sandal’, ‘Shirt’, ‘Sneaker’, ‘Bag’, ‘Ankle boot’)
- 下面通过tenssorflow加载该数据集,第一次加载可能稍慢,他要从官网上下载数据集。
1 | import pandas as pd |
运行的结果如下图所示:可以发现他们有10种不同的类别。
2.搭建三层神经网络
这里搭建神经网络就使用tensorflow中的keras进行搭建,下面详细讲解搭建的过程:
首先在纸上把整个神经网络的框架画出来:
由于每一张图片维度都是2828,所以输入层的单元数为28\28;因为是一张图片图像,像素比较多,故需要隐藏层维度大一点在此先设置为128,后面会说明隐藏层的大小对其结果的影响;输出层维度为10,因为有十个类别,最后可以通过softmax将其转化为10个概率,概率最大的那一个就是最佳的类别。
用keras中的顺序模型来搭建神经网络模型,模型是一层一层添加的,按照顺序。
1
model = tf.keras.Sequential()
添加输入层,使用Flatten(),将输入的二维向量转化为一位向量,即28*28的一维向量,作为输入单元数。
1
model.add(tf.keras.layers.Flatten(input_shape=(28,28)))
添加隐藏层,使用Dense()层,第一个参数为隐藏层维度,activation参数为激活函数,即通过激活函数传到下一层。
1
model.add(tf.keras.layers.Dense(128, activation='relu'))
常用的激活函数有:’relu’、’sigmoid’、’tanh’、’Leak relu’
sigmoid函数
Sigmoid函数是一个在生物学中常见的S型函数,也称为S型生长曲线。在信息科学中,由于其单增以及反函数单增等性质,Sigmoid函数常被用作神经网络的阈值函数,将变量映射到0,1之间。公式如下
函数图像如下
tanh函数
Tanh是双曲函数中的一个,Tanh()为双曲正切。在数学中,双曲正切“Tanh”是由基本双曲函数双曲正弦和双曲余弦推导而来。公式如下
函数图像如下
relu函数
Relu激活函数(The Rectified Linear Unit),用于隐层神经元输出。公式如下
函数图像如下
添加输出层,使用Dence()层,这里的激活函数就使用softmax函数,将其转化为概率输出。
1
model.add(tf.keras.layers.Dense(10, activation='softmax'))
至此神经网络模型搭建完毕,我们可以使用
model.summary()
来查看建立的模型的结构:1
2
3
4
5
6
7
8
9
10
11
12
13
14
15Model: "sequential"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
flatten (Flatten) (None, 784) 0
_________________________________________________________________
dense (Dense) (None, 128) 100480
_________________________________________________________________
dense_1 (Dense) (None, 10) 1290
=================================================================
Total params: 101,770
Trainable params: 101,770
Non-trainable params: 0
_________________________________________________________________
None 可以发现每一层的维度大小都有给出,包括每一层的参数个数paramters,这里隐藏层的参数为100480 = (784 + 1) 128,原因是前面的神经网络的介绍中已经说明,需要多加一个偏置单元,同理后面的1290 = (128 + 1) 10,也默认添加了一个偏置单元。
对该模型进行编译:
1
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['acc'])
optimizer:是优化方法,常用的优化方法有adam、SGD(随机梯度下降优化器)、RMSprop等;
loss:是损失函数,常用的有’mes’(均方差),这里由于是分类问题,使用的是前面所讨论的交叉熵损失函数,有两种’sparse_categorical_crossentropy’(适用于最后分类的顺序编码)、’categorical_crossentropy’(适用于最后分类是独热编码);
metrics=[‘acc’]可查看每一次的准确率;
用该模型训练数据集:
1
history = model.fit(train_image, train_lable, epochs=5,validation_data=(test_image,test_label))
epochs为训练代数,前面是训练集,后面是验证集。每次训练的结果都存在history里面,后面可以根据history将学习曲线以及准确率曲线画出来。
3.模型训练
下面是整个的完整代码,将学习曲线给画出来了。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import tensorflow as tf
(train_image, train_lable),(test_image, test_label) = tf.keras.datasets.fashion_mnist.load_data() # 加载fashion_mnist数据
def show_imgs(n_rows, n_cols, x_data, y_data, class_names):
assert len(x_data) == len(y_data)
assert n_rows * n_cols < len(y_data)
plt.figure(figsize = (n_cols * 1.4, n_cols * 1.6))
for row in range(n_rows):
for col in range(n_cols):
index = n_cols * row + col
plt.subplot(n_rows, n_cols, index+1)
plt.imshow(x_data[index], cmap='binary', interpolation = 'nearest')
plt.axis('off')
plt.title(class_names[y_data[index]])
plt.show()
def plot_learning_curves(history):
pd.DataFrame(history.history).plot(figsize=(8,5))
plt.grid(True)
plt.gca().set_ylim(0, 1)
plt.show()
class_names = ['T-shirt', 'Trouser', 'Pullover', 'Dress', 'Coat', 'Sandal', 'Shirt', 'Sneaker', 'Bag', 'Ankle boot']
show_imgs(3, 9, train_image, train_lable, class_names)
# 归一化
train_image = train_image/255
test_image = test_image/255
model = tf.keras.Sequential()
model.add(tf.keras.layers.Flatten(input_shape=(28,28))) # 28*28向量 将矩阵转化为向量
model.add(tf.keras.layers.Dense(128, activation='relu'))
model.add(tf.keras.layers.Dense(10, activation='softmax'))
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['acc'])
history = model.fit(train_image, train_lable, epochs=5,validation_data=(test_image,test_label))
plot_learning_curves(history)
通过输出的结果可以发现loss: 0.2982 - acc: 0.8909 - val_loss: 0.3448 - val_acc: 0.8773,准确度为0.89,测试集的准确度为0.87。
4.利用Dropout抑制过拟合
前面我们讨论过,一个神经网络越复杂,神经元越多,隐藏层越多拟合能力越强,但拟合的速度也越慢,更容易产生过拟合现象,那么为了抑制过拟合这里可以使用添加Dropout层抑制过拟合。
1
2
3
4
5model = tf.keras.Sequential()
model.add(tf.keras.layers.Flatten(input_shape=(28,28))) # 28*28向量 将矩阵转化为向量
model.add(tf.keras.layers.Dense(128, activation='relu'))
model.add(tf.keras.layers.Dropout(0.5)) # 添加Dropout层抑制过拟合
model.add(tf.keras.layers.Dense(10, activation='softmax'))就是在之前搭建的神经网络模型的基础上添加一层,参数为0.5代表抑制程度,Dropout的基本原理就是按照一定方法去掉一些神经元,使网络变简单,从而可以抑制过拟合。
5.使用回调函数
前面我们的学习曲线是根据训练完成后的history来画的,但当训练代数比较大时,我们需要时刻监控损失是否下降,以便更好得调整方法,即:在训练数据没有训练完成之前可以返回一些参数做别的操作,这就是回调函数的作用。
首先创建一个存放callbacks的文件夹,具体代码如下:
1
2
3
4logdir = './callbacks' # 如果报错改为绝对路径
if not os.path.exists(logdir):
os.mkdir(logdir)
output_model_file = os.path.join(logdir,'fashion_mnist_model.h5')这里只设置三种callbacks:TensorBoard、ModelCheckpoint、Earlyshopping;其他的callbacks请参考https://keras-cn.readthedocs.io/en/latest/other/callbacks/官方中文文档
TensorBoard:该回调函数是一个可视化的展示器
TensorBoard是TensorFlow提供的可视化工具,该回调函数将日志信息写入TensorBorad,使得你可以动态的观察训练和测试指标的图像以及不同层的激活值直方图。
如果已经通过pip安装了TensorFlow,我们可通过下面的命令启动TensorBoard:
tensorboard --logdir=/full_path_to_your_logs
EarlyStoping:当监测值不再改善时,该回调函数将中止训练
ModelCheckpoint:该回调函数将在每个epoch后保存模型到
filepath
filepath
可以是格式化的字符串,里面的占位符将会被epoch
值和传入on_epoch_end
的logs
关键字所填入
1 | logdir = './callbacks' |
下面是通过
tensorboard --logdir=callbacks
生成的本地网页数据:http://localhost:6006/
(通过端口6006在网页中打开)
6.(附)上一次用python实现的神经网络改为用Tensorflow2.0实现
1 | # -*- coding:utf-8 -*- |
可以发现代码量大大减少,并且得到了和之前相同的效果,并且在这里可以很方便的修改参数(隐藏层维度,隐藏层数量,激活函数等等)