1.删掉三个隐含层:
2.改用卷积神经网络来实现:
(1)因为卷积神经网络输入是图片,所以要先将x这个展开的一维的向量一个具有三通道的图片格式:
改成:
reshape之后还需要交换一下通道,perm就是通道的排列顺序:
这里的效果是把后两个挪到前面来, 前面的防盗后面去这样一个效果:
(2)进行第一个卷积层和pooling层:
激活函数是relu(第三个参数)
pooiling的步长最小得是2才能达到变小的效果
(3)同理再添加两层卷积层和池化层:
(4)再添加一个全连接层:
其他代码部分不变~~
3.最后的训练效果:
欧克~