Keras中自定义层/类在预测中的使用
srp phat
训练的时候:
class weightclip(Constraint):
def __init__(self, c=2):
self.c = c
def __call__(self, w):
return K.clip(w, -self.c, self.c)
def get_config(self):
return{'name':self.__class__.__name__, 'c':self.c}
预测的时候:
class weightclip:
def __in
原创
2021-01-04 16:51:58 ·
272 阅读 ·
0 评论