在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
是保存的模型文件名,可以自定義。使用這種方法保存模型可以保存模型的權重、結構和優化器的狀態,可以直接加載模型進行預測或繼續訓練。