下述操作都是基于:tensorflow-2.15.0
安装命令: pip install -i https://mirrors.aliyun.com/pypi/simple tensorflow
python下载:https://registry.npmmirror.com/binary.html?path=python/3.9.13/
TensorFlow是用数据流图做计算的,因此需要
先
创建一个
数据流图
。以一个简单的回归模型为例,其中与模型有关的元素由输入(Input)、重塑(Reshape)、ReLU层(ReLU Layer)、Logit层(Logit Layer)、Softmax、交叉熵(Crossentropy)、梯度(Gradient)、SGD训练(SGD Trainer)等部分。计算过程:
- 输入
- 重塑
- ReLU层会有两个参数,即 W h 1 W_{h1} Wh1和 b h 1 b_{h1} bh1,在
输出
前用ReLU激活函数做非线性处理。 - Logit层(输出),学习两个参数 W s m W_{sm} W
版权归原作者 WEL测试 所有, 如有侵权,请联系我们删除。