1 保存最佳权重
from keras.callbacks import ModelCheckpoint
filepath='weights.best.hdf5'
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=True, mode='max')
callbacks_list = [checkpoint]
model.fit(X, Y, validation_split=0.33, nb_epoch=150, batch_size=10,callbacks=callbacks_list, verbose=0)`
在模型训练前给定权重保存路径
save_best_only=True代表只保存最佳模型
在模型训练时,加入callbacks=callbacks_list即可。
后面调用时使用
from keras.models import Model
Model.load_weights(filepath1)
2 保存最佳模型
与保存最佳权重基本相同,只是需要吧文件名改成filepath='weights.best.h5’即可
后面调用时使用
from keras.models import load_model
model=load_model(filepath)