Kerasで早期打ち切り
Callback
関数内のEarlyStopping
を使用する。
マニュアルは下記
コールバック - Keras Documentation
呼び方
EarlyStopping(monitor='val_loss', min_delta=0, patience=0, verbose=0, mode='auto')
monitor
: 監視する値.min_delta
: 監視する値について改善として判定される最小変化値.patience
: 訓練が停止し,値が改善しなくなってからのエポック数
一般的な使い方
from keras.callbacks import EarlyStopping # Early-stopping early_stopping = EarlyStopping(patience=0, verbose=1) # training history = model.fit(X_train, Y_train, batch_size=batch_size, nb_epoch=nb_epoch, verbose=1, validation_split=0.1, callbacks=[early_stopping])