callbacks - EarlyStopping, ModelCheckpoint

1
2
3
4
5
6
7
8
9
10
11
12
13
14
from tensorflow.keras.callbacks import EarlyStopping, ModelCheckpoint

# earlystopping == 더이상 개선이 안될 경우 스탑하는 기능.
# 관심 가지는 변수 monitor = '변수'
# 연속해서 4번의 변화까지는 참는다(계속 진행한다) patience = 4
es = EarlyStopping(monitor='val_loss', mode='min', verbose=1, patience=4)

# 모델 임시 저장 modelcheckpoint
mc = ModelCheckpoint('best_model.h5', monitor='val_acc', mode='max', verbose=1, save_best_only=True)


# 콜백(callback) : 시스템이 어떤 상황이 되었을때 시스템에 의해 자동으로 호출되는 함수
model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['acc'])
history = model.fit(X_train, y_train, batch_size=128, epochs=2, callbacks=[es, mc], validation_data=(X_test, y_test))
Author

InhwanCho

Posted on

2022-12-01

Updated on

2022-12-01

Licensed under

Comments