230517 머신러닝 트러블(`lr` is deprected in Keras opt)

colab 사용 중 WARNING (`lr` is deprected in Keras opt)

문제

WARNING:absl:`lr` is deprecated in Keras optimizer, please use `learning_rate` or use the legacy optimizer, e.g.,tf.keras.optimizers.legacy.SGD.

 

과정

구글링한 결과

https://github.com/ageron/handson-ml2/issues/456

경고 문구가 뜨지만 아직은 구동이 된다.

 

원인

lr 인자는 Keras 2.3.0 버전에서 폐지되었다. 이 인자는 learning_rate 인자로 대체되었으며, 계속 사용할 수 는 있지만, 향후 버전에서는 제거될 수 있다. 따라서 learning_reate 인자를 사용 하는 것이 좋다.

 

 

해결

# 폐기된 방법
model.compile(loss='mean_squared_error', optimizer=SGD(lr=0.01))

# 새 방법
model.compile(loss='mean_squared_error', optimizer=SGD(learning_rate=0.01))