我有一个模型,想要为现有的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:使用序列设置数组元素。任何帮助将不胜感激