Как добавить метрики в модель керас - PullRequest
Новая
       17

Как добавить метрики в модель керас

0 голосов
/ 02 мая 2018

У меня есть модель, и я хочу добавить новые показатели rmse к существующим mae и r2. 

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=['mae', r2_keras, rmse]

ValueError: установка элемента массива с последовательностью. Любая помощь будет оценена

...