深度學習模型花費時間大多很長, 如果一次訓練過程意外中斷, 那么后續時間再跑就浪費很多時間. 這一次練習中, 我們利用 Keras checkpoint 深度學習模型在訓練過程模型, 我的理解是檢查訓練過程, 將好的模型保存下來. 如果訓練過程意外中斷, 那么我們可以加載最近一次的文件, 繼續進行訓練, 這樣以前運行過的就可以忽略.

那么如何 checkpoint 呢, 通過練習來了解.

• 數據: Pima diabete 數據
• 神經網絡拓撲結構: 8-12-8-1

1.效果提升檢查

如果神經網絡在訓練過程中, 其訓練效果有所提升, 則將該次模型訓練參數保存下來.

代碼:

# -*- coding: utf-8 -*- # Checkpoint NN model imporvements from keras.models import Sequential from keras.layers import Dense from keras.callbacks import ModelCheckpoint import numpy as np import urllib url = "http://archive.ics.uci.edu/ml/machine-learning-databases/pima-indians-diabetes/pima-indians-diabetes.data" raw_data = urllib.urlopen(url) dataset = np.loadtxt(raw_data, delimiter=",") X = dataset[:, 0:8] y = dataset[:, 8] seed = 42 np.random.seed(seed) # create model model = Sequential() model.add(Dense(12, input_dim=8, init='uniform', activation='relu')) model.add(Dense(8, init='uniform', activation='relu')) model.add(Dense(1, init='uniform', activation='sigmoid')) # compile model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy']) # checkpoint filepath = "weights-improvement-{epoch:02d}-{val_acc:.2f}.hdf5" # 中途訓練效果提升, 則將文件保存, 每提升一次, 保存一次 checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=True, mode='max') callbacks_list = [checkpoint] # Fit model.fit(X, y, validation_split=0.33, nb_epoch=150, batch_size=10, callbacks=callbacks_list, verbose=0) 

部分結果:

Epoch 00139: val_acc did not improve Epoch 00140: val_acc improved from 0.70472 to 0.71654, saving model to weights-improvement-140-0.72.hdf5 Epoch 00141: val_acc did not improve Epoch 00142: val_acc did not improve Epoch 00143: val_acc did not improve Epoch 00144: val_acc did not improve Epoch 00145: val_acc did not improve Epoch 00146: val_acc did not improve Epoch 00147: val_acc did not improve Epoch 00148: val_acc did not improve Epoch 00149: val_acc did not improve 

在運行程序的本地文件夾下, 我們會發現許多性能提升時, 程序自動保存的 hdf5 文件.

轉自：https://anifacc.github.io/deeplearning/machinelearning/python/2017/08/30/dlwp-ch14-keep-best-model-checkpoint/