Deep Learning
Learning rate 를 옵티마이저에서 셋팅하는 코드
yugyeong
2022. 12. 28. 15:48
Learning Rate (학습률)
딥러닝에서 학습률은 적당한 값을 넣어주는 것이 중요하다. 학습률에 따라 인공지능의 성능이 달라진다.
Learning rate 의 디폴트 값은 0.001 이고, 필요에 따라서 조금씩 조정을 하면 된다.
learning_rate를 옵티마이저에 셋팅할 때는, 아래처럼 작성하면 된다.
learning_rate : 학습률의 보폭을 설정, 기본 값은 0.001
model.compile(optimizer= Adam(learning_rate= 0.001), loss= 'mse', metrics= ['mse', 'mae'])