Keras 2.0 模型加载与保存

转 https://blog.csdn.net/yangzm/article/details/82685415

第一种

from keras.models import load_model
 
model.save('my_model.h5')  # creates a HDF5 file 'my_model.h5'
del model  # deletes the existing model
 
# returns a compiled model
# identical to the previous one
model = load_model('my_model.h5')

 

第二种

保存

# keras library import  for Saving and loading model and weights
from keras.models import model_from_json
from keras.models import load_model
 
# serialize model to JSON
#  the keras model which is trained is defined as 'model' in this example
model_json = model.to_json()
 
with open("model_num.json", "w") as json_file:
    json_file.write(model_json)

# serialize weights to HDF5
model.save_weights("model_num.h5")

加载

# load json and create model
json_file = open('model_num.json', 'r')
 
loaded_model_json = json_file.read()
json_file.close()
loaded_model = model_from_json(loaded_model_json)
 
# load weights into new model
loaded_model.load_weights("model_num.h5")
print("Loaded model from disk")
 
model.save('model_num.hdf5')
loaded_model=load_model('model_num.hdf5')