2017-12-07 10 views
0

説明が含まれているケラスモデル(簡単な文字列でもよい)を書いて、ファイルの内容をより簡単に把握できるようにしたいと思います。ケラスで説明付きのモデルを保存する

これを行う簡単な方法はありますか?私は、私がやりたいものを

model.save(filename) 
[...] 
model = keras.load_model(filename) 

でモデルを保存して読み込むことができ、これを達成するための最も簡単な方法であるものに

model.save(filename, description=string) 
[...] 
model, description = keras.load_model(filename) 

任意のアイデアの効果に何かありますか?

答えて

0

現在のところ、Kerasの現在のバージョンmodel.saveでこれを行う方法はありません。 loss functionmetricsload_modelなどのモデルの設定を保存するだけで、保存された内容が返されます。この問題に対する

一つ厄介な解決策は、手動Kerasにクラスmodels.pydescription変数を追加し、その結果、save_modelload_model関数にコードの必要な行を追加しています。繰り返しますが、これは恐ろしい解決策です!

関連する問題