如何将指标添加到模型keras

时间:2018-05-02 15:11:49

标签: python machine-learning neural-network keras metrics

我有一个模型,想要为现有的mae和r2添加新的指标rmse。

model = Sequential()
model.add(LSTM(
         input_shape=(n_past, nb_features),
         units=100,
         return_sequences=True))
model.add(Dropout(0.2))
model.add(LSTM(
          units=50,
          return_sequences=False))
model.add(Dropout(0.2))
model.add(Dense(units=1))
model.add(Activation("linear"))
model.compile(loss='mean_squared_error', optimizer='rmsprop', metrics=['mae', r2_keras])

指标的功能(非常类似于r2_keras的功能):

def rmse(y_true, y_pred):
    return sqrt(mean_squared_error( y_true, y_pred ))

所以只需将它添加到metrics数组中,但它会出错:

metrics=['mae', r2_keras, rmse]

ValueError:使用序列设置数组元素。任何帮助将不胜感激

0 个答案:

没有答案