tf.losses.sparse_softmax_cross_entropy
-
[tensorflow] How to use tf.losses.sparse_softmax_cross_entropy() on tensorflow 2.xAI 2020. 3. 25. 18:44
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_)