起因
自己寫了一個f1_score
的函數作爲metrics中的一項,然後保存了(模型+參數)。load_model(model_path)
·的時候出了問題。找不到f1_score
這個函數。
方案
- 加載模型的時候
load_model(self.saved_model, custom_objects={"f1_score ": f1_score })
加上引用
後加,其實模型和參數不用分開保存
是因爲出了這個問題:Keras使用 Lambda後訓練出的模型加載後,預測結果爲隨機
自己寫了一個f1_score
的函數作爲metrics中的一項,然後保存了(模型+參數)。load_model(model_path)
·的時候出了問題。找不到f1_score
這個函數。
load_model(self.saved_model, custom_objects={"f1_score ": f1_score })
加上引用是因爲出了這個問題:Keras使用 Lambda後訓練出的模型加載後,預測結果爲隨機