Keras提供了简单的方法来保存和加载模型。可以使用model.save()
方法来保存模型,使用keras.models.load_model()
方法来加载模型。
保存模型:
model.save('my_model.h5')
加载模型:
from keras.models import load_model
model = load_model('my_model.h5')
注意:保存的模型文件会包含模型的架构、权重和优化器的配置。因此,加载模型时不需要再重新编译模型。