Блог

Я создал пользовательское ограничение для своей модели 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
 

Может кто-нибудь, пожалуйста, помочь?
Заранее спасибо!