Kerasでモデルを保存およびロードする方法を教えてください。

Kerasでは、モデルを保存するためにmodel.save()メソッドを使用し、モデルをロードするためにkeras.models.load_model()メソッドを使用します。具体的な手順は以下の通りです。

モデルの保存:

# 保存模型
model.save('my_model.h5')

モデルをロードする:

from keras.models import load_model

# 加载模型
model = load_model('my_model.h5')

上記のコードでは、my_model.h5という名前の保存されたモデルファイルがあり、任意の名前を選択できます。この方法を使用すると、モデルの重み、構造、および最適化器の状態が保存され、モデルを直接ロードして予測を行うか、トレーニングを継続することができます。

コメントを残す 0

Your email address will not be published. Required fields are marked *


广告
広告は10秒後に閉じます。
bannerAds