728x90
callbacks
중간에 일찍 끝내야 할때 과적합 방지 그리고 중간에 모델 저장 이정도면 충분하다 callbacks =[끝내기,저장]
이런식으로 monitor 에는 손실함수? 를 사용해준다
from keras.callbacks import EarlyStopping,ModelCheckpoint
cp=ModelCheckpoint("best.ckpt",
save_weights_only=True,
save_best_only=True,
monitor="val_loss",
verbose=1)
el=EarlyStopping(monitor="val_loss",patience=3)
h=model.fit(train_g,
validation_data=val_g,
epochs=100,
callbacks=[el,cp])
'python > ML,DL' 카테고리의 다른 글
python tensorflow gpu 사용 설정 (0) | 2023.05.21 |
---|
댓글