Keras提供了簡單的方法來保存和加載模型。可以使用model.save()
方法來保存模型,使用keras.models.load_model()
方法來加載模型。
保存模型:
model.save('my_model.h5')
加載模型:
from keras.models import load_model
model = load_model('my_model.h5')
注意:保存的模型文件會包含模型的架構、權重和優化器的配置。因此,加載模型時不需要再重新編譯模型。