温馨提示×

Keras模型如何保存和加载

小樊
93
2024-06-18 13:07:54
栏目: 深度学习

Keras模型可以通过调用model.save()方法保存为HDF5格式的文件,以便后续加载和重用。加载模型时,可以使用keras.models.load_model()方法加载保存的模型文件。

保存模型示例:

from keras.models import load_model

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

加载模型示例:

from keras.models import load_model

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

0