代码之家  ›  专栏  ›  技术社区  ›  deltascience

当目标有上限和下限时,在回归任务中使用什么激活函数

  •  0
  • deltascience  · 技术社区  · 7 年前

    我正在尝试使用keras执行回归任务,目标是0到15之间的天数。

    def rmse(y_true, y_pred):
        return backend.sqrt(backend.mean(backend.square(y_pred - y_true), axis=-1))
    
    
    model = Sequential()
    model.add(Dense(1276, input_dim=1276, kernel_initializer='normal', activation='relu'))
    model.add(Dense(200, kernel_initializer='normal', activation='relu'))
    model.add(Dense(100,kernel_initializer='normal' ,  activation='relu'))
    model.add(Dense(1, activation="linear"))
    # Compile model
    optimizer = optimizers.Adam(lr=1e-4)
    model.compile(loss=rmse, optimizer=optimizer)
    # Fit the model
    model.fit(X,y , validation_split=0.33, epochs=100, batch_size=512)
    

    当我训练这个模型时,我得到了非常大的值。我想知道我的实现中是否遗漏了一些东西。我应该为此类问题使用特定的激活功能吗?

    0 回复  |  直到 7 年前
    推荐文章