tf 多层感知机(MLP)与线性模型

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Eric_LH/article/details/82529892

多层感知机与线性模型类似,所不同的地方在于层数增加了,以及引入了非线性激活函数(这里使用了relu)。

该模型输入一个向量(比如这里是拉直的 1×784 手写体数字图片),输出 10 维的信号,分别代表这张图片属于 0到 9 的概率。这里我们加入了一个 predict 方法,对图片对应的数字进行预测。在预测的时候,选择概率最大的数字进行预测输出。

猜你喜欢

转载自blog.csdn.net/Eric_LH/article/details/82529892
tf