機械学習・自然言語処理の勉強メモ

学んだことのメモやまとめ

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])