新聞中心
這里有您想知道的互聯(lián)網(wǎng)營銷解決方案
keras保存最佳訓(xùn)練模型的方法-創(chuàng)新互聯(lián)
這篇文章主要講解了keras保存最佳訓(xùn)練模型的方法,內(nèi)容清晰明了,對此有興趣的小伙伴可以學(xué)習(xí)一下,相信大家閱讀完之后會有幫助。
1、只保存最佳的訓(xùn)練模型
2、保存有所有有提升的模型
3、加載模型
4、參數(shù)說明
只保存最佳的訓(xùn)練模型
from keras.callbacks import ModelCheckpoint filepath='weights.best.hdf5' # 有一次提升, 則覆蓋一次. checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1,save_best_only=True,mode='max',period=2) callbacks_list = [checkpoint] model.compile(loss='categorical_crossentropy', optimizer=optimizers.Adam(lr=2e-6,decay=1e-7),metrics=['acc']) history1 = model.fit_generator( train_generator, steps_per_epoch=100, epochs=40, validation_data=validation_generator, validation_steps=100, callbacks=callbacks_list)
分享文章:keras保存最佳訓(xùn)練模型的方法-創(chuàng)新互聯(lián)
轉(zhuǎn)載來于:http://ef60e0e.cn/article/dhopjp.html