Tensorflow实战(黄文坚)——多层感知机代码实现

刚入门tensorflow,最近在看黄文坚的《Tensorflow实战》,看到“Tensorflow实现多层感知机”这节,就手动把代码实现了一下。
导入mnist数据集的时候,不知道为什么总是很慢很慢,一直运行不停,所以就把数据集下载到本地了。网上有人提供云盘链接,搜一下就好。
下面代码中 ‘./MNIST_data/’ 是我电脑中mnist的路径(下载的时候自己选择),最前面的 ‘.’ 表示当前目录(如果是两个点,表示当前目录的上一级)。后面的 ‘/’ 表示读取该文件夹下的所有文件。注意,下载好的4个压缩文件不用解压缩!!!运行read_data_sets的时候会自动解压的。
在这里插入图片描述

from tensorflow.examples.tutorials.mnist import input_data
import tensorflow as tf
mnist=input_data.read_data_sets('./MNIST_data/',one_hot=True)

##########构建神经网络第一步,前向建立计算图##########

#设置参数Variable并进行初始化
    #输入层和hidden layer1的神经元数量分别为784和300
in_units=784
h1_units=300
    #权重和偏置的初始化
W1=tf.Variable(tf.truncated_normal([in_units,h1_units],stddev=0.1))
b1=tf.Variable(tf.zeros([h1_units]))
W2=tf.Variable(
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
抱歉,由于时间和篇幅的限制,这个问题无法在一个简短的回答中得到充分解答。但是,我可以给你提供一些参考资料和指导,帮助你开始编写基于二维灰度图的轴承故障诊断的TensorFlow框架程序。 首先,你需要收集足够的轴承故障图像数据集,并将其划分为训练集、验证集和测试集。然后,你需要对这些图像进行预处理,比如缩放、裁剪、归一化等操作,以便于模型的训练和预测。 接下来,你需要选择适合你的问题的深度学习模型,比如卷积神经网络(CNN)。你可以使用TensorFlow中的高级API,例如Keras或Estimator来构建和训练你的模型。你需要定义模型的架构、损失函数和优化器,并使用训练集进行模型的训练和验证集进行模型的评估和调整。 最后,你可以使用测试集对你的模型进行测试和评估,并根据结果进行模型的进一步改进和优化。 以下是一些有用的参考资料: 1. TensorFlow官方文档:https://www.tensorflow.org/ 2. TensorFlow中文社区:https://tensorflow.google.cn/ 3. TensorFlow模型库:https://github.com/tensorflow/models 4. 《Python深度学习》(Francois Chollet):https://book.douban.com/subject/27087503/ 5. 《TensorFlow实战Google深度学习框架》(黄文坚、唐源):https://book.douban.com/subject/26976457/ 6. 《深度学习入门之TensorFlow》(张志华):https://book.douban.com/subject/27087503/

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值