활연개랑

[tensorflow] 오류 해결 :ValueError: Shapes (5,) and (5, 5) are incompatible 본문

딥러닝

[tensorflow] 오류 해결 :ValueError: Shapes (5,) and (5, 5) are incompatible

승해tmdhey 2022. 5. 28. 17:56
반응형

classification 딥러닝을 training 시키려는데 다음과 같은 오류가 발생하였습니다.

ValueError: Shapes (5,) and (5, 5) are incompatible

저한테 오류가 났던 이유는 Accuracy 측정용으로 아래와 같이 SparseCategoricalAccuracy를 사용하였는데,

tf.keras.metrics.SparseCategoricalAccuracy()

loss로는 그냥 categorical cross entropy를 사용했기 때문이었습니다.

 

따라서 loss를 아래와 같이 SparseCategoricalCrossentropy를 사용해주면, 오류를 해결할 수 있습니다.

tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True,reduction=tf.keras.losses.Reduction.NONE)