keras 保存完整模型,加载模型出错Unknown metric function

起因

自己写了一个f1_score的函数作为metrics中的一项,然后保存了(模型+参数)。load_model(model_path)·的时候出了问题。找不到f1_score这个函数。

方案
  1. 加载模型的时候load_model(self.saved_model, custom_objects={"f1_score ": f1_score }) 加上引用
后加,其实模型和参数不用分开保存

是因为出了这个问题:Keras使用 Lambda后训练出的模型加载后,预测结果为随机

解答

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章