我对经过回归训练的神经网络输出层的激活函数有点困惑。在大多数教程中,输出层使用“sigmoid”将结果恢复到 0 到 1 之间的一个不错的数字。
但是在TensorFlow网站上的这个初学者示例中,输出层根本没有激活函数?这是允许的吗?结果不会是一个到处都是疯狂的数字吗?或者也许 TensorFlow 有一个隐藏的默认激活?
此代码来自您根据汽车马力预测每加仑英里数的示例。
// input layer
model.add(tf.layers.dense({inputShape: [1], units: 1}));
// hidden layer
model.add(tf.layers.dense({units: 50, activation: 'sigmoid'}));
// output layer - no activation needed ???
model.add(tf.layers.dense({units: 1}));