Keras提供了簡(jiǎn)單的方法來(lái)保存和加載模型?？梢允褂?code>model.save()方法來(lái)保存模型，使用keras.models.load_model()方法來(lái)加載模型。
保存模型：
model.save('my_model.h5')
加載模型：
from keras.models import load_model
model = load_model('my_model.h5')
注意：保存的模型文件會(huì)包含模型的架構(gòu)、權(quán)重和優(yōu)化器的配置。因此，加載模型時(shí)不需要再重新編譯模型。
億速云公眾號(hào)
手機(jī)網(wǎng)站二維碼