Keras Lambda Layer для расчета температуры в модели LSTM - PullRequest
Новая
       26

Keras Lambda Layer для расчета температуры в модели LSTM

0 голосов
/ 20 января 2019

У меня проблема с Keras Lambda Layers. Я использую Keras функционал API для создания моей модели. Кроме того, я хотел бы сохранить модель в файл h5.

Отлично работает с этим кодом: 

shared_layer = LSTM(size, return_sequences=True, dropout=self.params['dropout'][0], recurrent_dropout=self.params['dropout'][1])(shared_layer)
output = shared_layer
output = TimeDistributed(Dense(n_class_labels, activation='softmax'), name=modelName+'_softmax')(output)

Но если я добавлю следующую строку, я получаю ошибку при сохранении модели. Хотя это не выдает ошибку при обучении: 

logits_temperature = Lambda(lambda x : x / self.params['temperature'])(output)
output = TimeDistributed(Dense(n_class_labels, activation='softmax'), name=modelName+'_softmax')(logits_temperature)

Вот ошибка, которую я получаю при сохранении:

TypeError: cannot serialize '_io.TextIOWrapper' object

...