получить имя функции потерь, используемой в модели keras - PullRequest
Новая
       2

получить имя функции потерь, используемой в модели keras

1 голос
/ 03 марта 2020

Я хочу сохранить название функции потерь, которую я использовал в моей модели керас. Я заглянул в документацию , но не нашел способа получить это имя. Если возможно, я также хочу сохранить это имя на случай, если я использую пользовательскую функцию потери. Или, по крайней мере, извлечь информацию из модели, в которой я использовал пользовательскую функцию потерь. Вот как выглядит моя модель: 

model = Sequential()
model.add(Dense(5, input_dim=4, activation='tanh'))
model.add(Dense(5, activation='tanh'))
model.add(Dense(5, activation='tanh'))
model.add(Dense(3))
model.compile(loss='mean_squared_error', optimizer='nadam', metrics=['accuracy'])

А для нестандартной потери: 

model.compile(loss=custom_loss, optimizer='nadam', metrics=['accuracy'])
...