데이터과학 삼학년

텐서플로우에도 파이토치의 top_k처럼 모델이 output으로 내뱉는 결과중 상위 k개를 뽑아내는 함수가 있다. tf.math.top_k(output, k) 이 것을 통하면 prob과 indices를 한번에 확인 가능하다!!! import tensorflow as tf # 라벨 딕셔너리 정의 label_dict = {0: 'cat', 1: 'dog', 2: 'bird'} # 가장 높은 값과 인덱스 찾기 values, indices = tf.math.top_k(output, k=1) print("Predicted index:", indices.numpy()[0])print("Predicted value:", values.numpy()[0]) @tf.function(input_signature=[tf.Ten..