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'])