Keras导入自定义metric模型报错: unknown metric function: Please ensure this object is passed to`custom_object‘_please ensure this object is passed to the `custom

Keras的模型自定义了metric或者loss,
在保存成后h5的时候没有问题, 但是在使用load_model导入的时候却会报错:

unknown metric function: HammingScore. Please ensure this object is passed to the custom_objects argument.

这是因为自定义的参数没有传递进去. 有两个解决办法:

1. 如果只需要做预测, 不再训练, 则可以直接加compile = False:

model = keras.models.load_model('model.h5', compile = False)
1

如果再需要训练或者修改, 就把自己定义的metrics代码加上, 并且再次编译:

model.compile(loss='binary_crossentropy',
              optimizer=Ada,
              metrics=[HammingScore]) # 这里HammingScore是我自定义的metric
1
2
3

2. 在导入的时候把自定义的metric/loss作为键值传入custom_objects :

model = keras.models.load_model('model.h5', custom_objects={'HammingScore': HammingScore} )
1

注意键值要一致才可以.