【Keras】load_modelで自作の関数を読み込む

スポンサーリンク
python
スポンサーリンク

今回は、自作の関数を持つモデルをKerasのload_modelで読み込む方法を解説します。

Kerasチュートリアルのmnistモデルで試してみます。

自作の関数としてrelu6を作成し、reluと置き換えています。

学習させたモデルを保存します。

次に、load_modelでモデルを読み込んでみます。

すると、次のようなエラーが出てきました。

Unknown activation function:relu6

自作の関数であるrelu6が読み込めていないようです。

このような場合には、次のように書くと読み込めます。

load_modelの引数に、cutom_objectsを設定し、読み込みたい自作関数を指定してください。

タイトルとURLをコピーしました