Как определить мою собственную функцию потерь в Керасе? - PullRequest
Новая
       25

Как определить мою собственную функцию потерь в Керасе?

0 голосов
/ 11 июня 2018

Я пытаюсь использовать CuDNNLSTM в Keras для решения некоторых последовательных вопросов с несколькими выходами.Мой код написан следующим образом: 

inputs = Input(...)
lstm_outputs = CuDNNLSTM(n_neurons, return_sequences=True)(inputs)
outputs = TimeDistributed(Dense(3))(lstm_outputs)
model = Model(inputs=inputs, outputs=outputs)

Когда я использую mean_square_error в качестве функции потерь в Keras, результаты выглядят как

error = np.mean (np.square (y_pred - y_real))

где номер функции y_pred и y_real равен 3.

Но я хочу определить функцию потерь следующим образом:

error = np.mean (np.square (y_pred [:,:, 0] - y_real [:,:, 0]) + np.square (y_pred [:,:, 1] - y_real [:,:, 1]) + np.square (y_pred [:,:, 2] - y_real [:,:, 2]))

Я не знаю, как это реализовать, может кто-нибудь дать мне немногопредложения?Большое спасибо!

...