오뚝이개발자

[Keras] 케라스 AttributeError: module 'keras.optimizers' has no attribute 'RMSprop' 에러 해결법 본문

AI/AI 개발

[Keras] 케라스 AttributeError: module 'keras.optimizers' has no attribute 'RMSprop' 에러 해결법

땅어 2021. 10. 27. 20:59
728x90
300x250

 

 

케라스에서 optimizer로 RMSprop을 사용하다 보면 위와 같은 에러가 나오는 경우가 많다. 아마 버전 문제이거나 import 형식 때문인 것 같은데(구체적인 원인은 잘 모르겠다...) 하지만 해결법은 알고 있음!!! ㅎㅎ

 

이런 경우 대부분 아래와 같이 사용했을 경우가 많다.

import keras

model = ~~~~....	# 대충 이 부분에 model definition을 했다고 한다면

model.compile(loss='categorical_crossentropy', 
              optimizer=RMSprop(lr=0.001), 
              metrics=['accuracy'])

 

이 때 실행을 돌리면 AttributeError: module 'keras.optimizers' has no attribute 'RMSprop' 와 같은 에러가 나올 때가 있다. 이럴 땐 아래와 같이 import 방식을 바꿔주면 해결된다.

from tensorflow.keras import optimizers

 

 

 

728x90
300x250
Comments