📝딥러닝 텐서플로우 옵티마이저 런닝레이트 셋팅하기
인공지능 컴파일할때 아래 코드처럼 문자열이 아닌 함수로 불러오면 사용할 옵티마이저의 런닝레이트를 지정이 가능하다
※ 디폴트 런닝레이트는 0.001
임포트해서 함수를 불러와 셋팅도 가능 (from tensorflow.keras.optimizers import Adam)
def build_model() :
model = Sequential()
model.add( Dense( units= 64, activation= 'relu', input_shape= (9,) ) )
model.add( Dense( units= 64, activation= 'relu') )
model.add( Dense( units=1, activation= 'linear' ) )
# learning_rate을 셋팅하고 싶다면 문자열말고 tf.keras.optimizers.로 옵티마이저를 불러와 설정
model.compile( optimizer= tf.keras.optimizers.Adam(learning_rate=0.001), loss= 'mse', metrics= [ 'mse', 'mae' ])
return model
learning rate란 미분 기울기의 이동 보폭(step)
'인공지능 > Deep Learning' 카테고리의 다른 글
[Deep Learning] TensorFlow EarlyStopping, Callback (0) | 2022.12.28 |
---|---|
[Deep Learning] 학습중에 평가하기 validation (0) | 2022.12.28 |
[Deep Learning] epochs history 차트 (0) | 2022.12.28 |
[Deep Learning] DNN TensorFlow Regression 수치예측 문제 모델링 (1) | 2022.12.28 |
[Deep Learning] TensorFlow GridSearch (0) | 2022.12.27 |
댓글