小编Pop*_*rin的帖子

为什么 ReLU 用于神经网络的回归?

我正在关注带有 Keras 教程的官方 TensorFlow,但我被困在这里:预测房价:回归 - 创建模型

为什么将激活函数用于预测连续值的任务?

代码是:

def build_model():
    model = keras.Sequential([
        keras.layers.Dense(64, activation=tf.nn.relu, 
                   input_shape=(train_data.shape[1],)),
        keras.layers.Dense(64, activation=tf.nn.relu),
        keras.layers.Dense(1)
    ])

    optimizer = tf.train.RMSPropOptimizer(0.001)

    model.compile(loss='mse', optimizer=optimizer, metrics=['mae'])
    return model
Run Code Online (Sandbox Code Playgroud)

regression machine-learning neural-network keras activation-function

5
推荐指数
1
解决办法
4849
查看次数