#python #keras #model #tf.keras
Вопрос:
Я создал пользовательское ограничение для своей модели keras, и я смог обучить модель, сделать прогноз и сохранить модель. Пользовательское ограничение выглядит следующим образом:
class WeightClip(Constraint):
def __init__(self, c):
self.c = c
def __call__(self, p):
return tf.keras.backend.clip(p, -self.c, self.c)
def get_config(self):
return {'name': self.__class__.__name__,
'c': self.c}
Я сохраняю модель с помощью:
model.save("model.h5")
При загрузке модели с помощью:
model = load_model("model.h5")
Я получаю сообщение об ошибке:
ValueError: Unknown constraint: WeightClip
Может кто-нибудь, пожалуйста, помочь?
Заранее спасибо!