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