본문 바로가기
Data Science/Deep Learning

Keras ReduceLROnPlateau 클래스

by Queen2 2022. 11. 15.
728x90

지정한 metric이 향상이 안될 때, 학습률을 낮추도록 조정하는 기능을 수행함

patience로 지정한 횟수의 epoch 만큼 향상이 없을 때를 기준으로 함

 

tf.keras.callbacks.ReduceLROnPlateau(
    monitor="val_loss",
    factor=0.1,
    patience=10,
    verbose=0,
    mode="auto",
    min_delta=0.0001,
    cooldown=0,
    min_lr=0,
    **kwargs
)

 

 

 

Source:

https://keras.io/api/callbacks/reduce_lr_on_plateau/

728x90
반응형

댓글