본문 바로가기
오류

[오류 해결] AttributeError: 'Sequential' object has no attribute 'predict_classes'

by Leunco 2021. 10. 24.

tensorflow 버전 2.6이후로 predict_classes가 없기 때문에 발생하는 오류라고 한다. 

따라서 predict_classes 대신 다음 코드로 대체하면 된다.

# 오류 발생
predicted = model.predict_classes(token_list, verbose=0)

# 오류 해결
y_prob = model.predict(token_list, verbose=0) 
predicted = y_prob.argmax(axis=-1)

 

출처 : https://stackoverflow.com/questions/68776790/model-predict-classes-is-deprecated-what-to-use-instead

반응형

댓글