Keras에서 예측에 일괄 처리 크기가 필요한 이유는 무엇입니까? Keras에서

Question

, predict_classes()를 사용하여 datatest가 예측하는 :

classes = model.predict_classes(X_test, batch_size=32)

나는 훈련 batch_size의 사용은 알고 있지만, 왜 예측을위한 batch_size 필요합니까? 어떻게 작동합니까?

Answer 1

Keras는 100 개 요소의 벡터를 입력하는 것처럼 여러 값을 동시에 예측할 수 있습니다. Keras는 각 요소에 대해 하나의 예측을 계산하여 100 개의 출력을 제공합니다. 이 계산은 batch_size로 정의 된 일괄 처리로 수행 할 수도 있습니다.

이것은 CPU/GPU RAM에 모든 데이터를 동시에 넣을 수없고 배치 처리가 필요한 경우를위한 경우입니다.