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後訓練出的模型加載後,預測結果爲隨機

解答

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