load model實(shí)現(xiàn)加載含有參數(shù)自定義類的方法

本篇文章展示了load model實(shí)現(xiàn)加載含有參數(shù)自定義類的方法具體操作，代碼簡明扼要容易理解，絕對能讓你眼前一亮，通過這篇文章的詳細(xì)介紹希望你能有所收獲。

Keras在代碼結(jié)構(gòu)上由面向?qū)ο蠓椒ň帉?，完全模塊化并具有可擴(kuò)展性，其運(yùn)行機(jī)制和說明文檔有將用戶體驗(yàn)和使用難度納入考慮，并試圖簡化復(fù)雜算法的實(shí)現(xiàn)難度 。

加載自定義模型和函數(shù)，如下圖

這個SelfAttention層是在訓(xùn)練過程自己定義的一個class，但如果要加載這個自定義層，需要在load_model里添加custom_objects字典，這個自定義的類，不要用import ，最好是直接復(fù)制進(jìn)再訓(xùn)練的模型中。

如果直接運(yùn)行上面的代碼，會出現(xiàn)一個init初始化錯誤，如下圖，

再來看看 這個SelfAttention 自定義的類的初始化

這就說明再調(diào)用這個類的時候，輸入的ch=256并不會初始化這個類，需要先自定義好初始化值，如下圖

調(diào)用方式不變

這樣問題就解決啦！

補(bǔ)充知識：keras load model的時候，報錯('Keyword argument not understood:', u'******')如何解決

由于keras不同版本的API有變化，因此在一個keras版本下訓(xùn)練的模型在另一個keras版本下加載時，可能會出現(xiàn)諸如('Keyword argument not understood:', u'data_format')等報錯。

通過打開*.h6文件，查看該模型訓(xùn)練所用環(huán)境，再安裝該環(huán)境即可解決報錯。

查看Keras Model所用的Keras環(huán)境的方法

import h6py

f = h6py.File('Model.h6', 'r')
print(f.attrs.get('keras_version'))

根據(jù)輸出的keras版本安裝對應(yīng)版本的keras即可解決加載問題。

看完上述內(nèi)容，你們掌握load model實(shí)現(xiàn)加載含有參數(shù)自定義類的方法了嗎？如果還想學(xué)到更多技能或想了解更多相關(guān)內(nèi)容，歡迎關(guān)注億速云行業(yè)資訊頻道，感謝各位的閱讀！