[tensorflow] How to use tf.losses.sparse_softmax_cross_entropy() on tensorflow 2.x

 

 tensorflow 1.4에서 예제를 실습하다.

버전을 2.1로 업데이트 했다.

그러고 나니 실습중이던 예제에 에러가 생기기 시작했다..

 

---

 

아래 코드는 y값과 pred값의 loss를 구하는 함수이다.

tf.losses.sparse_softmax_cross_entropy(labels=y, logits=y_)

tensorflow 2.x 에서는 위의 함수를 사용할 수 없다.

그래서 대체 함수를 찾아야했다.

함수를 찾기전

다양한 문제들로 돌아돌아돌아..

저 함수의 문제라는걸 판단할 수 있었다.

 

 

이제 아래 함수를 사용하면 된다. 참조

cce = tf.keras.losses.SparseCategoricalCrossentropy()
cce(y, y_)