본문 바로가기
python/ML,DL

하이퍼 파라미터

by 1.5볼트 2023. 5. 29.
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

댓글