기타

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

킹남지 2022. 6. 27. 03:19
반응형

한참 동안 torch만 사용하다, Keras를 사용했는데 오류가 발생했다.

내용 그대로 tensorflow 2.6 이후부터는 predict_classes 를 지원하지 않는다.

 

다중 분류 문제에서는 아래와 같이,

predict_x = model.predict(X_test) 
classes_x = np.argmax(predict_x,axis=1)

그리고 이진 분류 문제에서는 아래와 같이 대체해 주면 된다.

(predict_x > 0.5).astype("int32")
반응형