Tensorflow构建一个神经网络

109 阅读1分钟

下面是一个要构建的神经网络:
三层神经网络,第一层有25个units,第二层有15个units,输出层一个unit

image.png

注意:

当使用Tensorflow时,按照惯例,没有显式的将变量分解为层,当说模型是几个层串在一起的顺序模型时,通常只是在Dense()中给出units和activation。

步骤1.通过如下四行代码构建出神经网络模型

image.png

步骤2.下面五行代码帮助我们训练出合适的模型,并实现预测

image.png