[Troubleshooting] ValueError: Shapes (None, 1) and (None, 2) are incompatible

문제 개요 :

전이 학습을 진행하던 도중 class의 개수가 여러 개가 있는 모델을 학습시키다가 binary 학습을 시키다가 발생하였습니다.

 

에러 내용 :

해결 방법 :

시도해 볼 수 있는 다른 해결 방안도 있을 것으로 생각되지만 내가 해결한 방법을 제시해 보겠습니다.

 

- 문제가 발생할 수 있는 모습

activation 함수를 categorical에서 사용한대로 softmax를 사용하게 되면 위와 같은 에러를 발생될 수 있습니다.

그리고 다음으로 model.compile 함수에서 loss 내용이 categorical_crossentropy로 되어있으면 문제가 발생할 수 있습니다.

 

- 해결 한 방법 

1. activation 함수는 sigmoid를 사용하여 바이너리 분석에 맞게 변경

2. model.compile 함수에서 loss 내용을 sparse_categorical_crossentropy로 변경