파이썬) [model] predict 함수

1. predict 함수 소개

입력 샘플에 대한 출력 예측을 생성한다.

predict(
    x,
    batch_size=None,
    verbose='auto',
    steps=None,
    callbacks=None,
    max_queue_size=10,
    workers=1,
    use_multiprocessing=False
)

 

2. predict 함수 매개 변수 및 반환 값

1) 매개 변수

x	입력 샘플 - Numpy 배열(또는 배열 유사) 또는 배열 목록 - TensorFlow 텐서 또는 텐서 목록
batch_size	정수 또는 None.  배치당 샘플 수. 지정 하지 않으면 기본값은 32
verbose	"auto", 0, 1 또는 2  0 = 무음, 1 = 진행률 표시줄, 2 = 한 줄.  "auto"대부분의 경우 기본값은 1
steps	예측 라운드가 완료되었다고 선언하기 전의 총 단계(샘플 배치) 수
callbacks	예측 중에 적용할 콜백 목록
max_queue_size	생성기 대기열의 최대 크기 지정하지 않으면 기본값은 10
workers	프로세스 기반 스레딩을 사용할 때 스핀업할 최대 프로세스 수지정하지 않으면 workers기본값은 1
use_multiprocessing	부울. 생성기 또는 keras.utils.Sequence입력에만 사용

 

2) 반환

예측의 Numpy 배열

예를 들어 10개의 카테고리가 있다면 여기에 상응하는 모델의 신뢰도가 배열로 보여진다.

array([0.              , 0.              , 0.              , 0.              , 0.              ,
          0.1025838, 0.              , 0.1096932, 0.              , 0.7877327],
          dtype=float32)  

 

<참고 자료>

https://www.tensorflow.org/api_docs/python/tf/keras/Model#predict

tf.keras.Model  |  TensorFlow Core v2.9.1