keras加載模型時(shí)有自定義metrics、loss時(shí)出現(xiàn)ValueError: Unknown metric function:***的解決方法

在使用keras時(shí)經(jīng)常會(huì)使用到存儲(chǔ)模型和加載模型。在存儲(chǔ)時(shí)使用

model.save("model.h5")

如果此時(shí)存儲(chǔ)的model中有自定義的loss或者metrics，如果使用model = keras.models.load_model("model.h5")在加載模型的時(shí)候會(huì)報(bào)錯(cuò)：

ValueError: Unknown metric function:acc_top5

這是因?yàn)樽远x的函數(shù)沒(méi)有被保存，加載出來(lái)就會(huì)報(bào)錯(cuò)，解決方案為：

model = keras.models.load_model("model.h5", custom_objects={'acc_top5': acc_top5})

如果是自定義loss，則custom_objects={}括號(hào)中做對(duì)應(yīng)修改：

model = keras.models.load_model("model.h5", custom_objects={'loss_fun': loss_fun})
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點(diǎn)，簡(jiǎn)書(shū)系信息發(fā)布平臺(tái)，僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容