TensorFlow之双隐含层多层感知器（MLP）

最新推荐文章于 2024-08-12 07:02:19 发布

丶kaka

最新推荐文章于 2024-08-12 07:02:19 发布

阅读量1.8k

点赞数

分类专栏：深度学习文章标签： TensorFlow 多层感知器 mlp 双隐含层

本文链接：https://blog.csdn.net/u012348655/article/details/76099631

版权

本文介绍了如何在TensorFlow中修改程序，实现一个带有双隐含层的多层感知器（MLP）。通过确保第二层隐含层的权重初始化与第一层相似，解决了模型正确率低的问题，最终达到约98%的正确率。

摘要由CSDN通过智能技术生成

程序改自上一篇博客，使用了双隐含层，第二层隐含层初始w需要和第一层类似，否则程序正确率一直在0.1左右。修改后的程序正确率也在98%左右。

# -*- coding:utf-8 -*-
from tensorflow.examples.tutorials.mnist import input_data
import tensorflow as tf

# data
dir='/home/kaka/Documents/input_data'
mnist = input_data.read_data_sets(dir, one_hot=True)

# model
sess = tf.InteractiveSession()

hd1in_units = 784
hd1out_units = 500
hd2out_units = 300

w1 = tf.Variable(tf.truncated_normal([hd1in_units, hd1out_units], stddev=