Kerasでは、データテストのクラスを予測するために、predict_classes()
が使用されます。
例えば:
classes = model.predict_classes(X_test, batch_size=32)
私の質問は、batch_size
トレーニング中ですが、なぜbatch_size
予測のために?どのように機能しますか?
Kerasは複数の値を同時に予測できます。たとえば、100要素のベクトルを入力した場合、Kerasは各要素に対して1つの予測を計算し、100の出力を生成できます。この計算は、batch_sizeで定義されたバッチで実行することもできます。
これは、CPU/GPU RAMにすべてのデータを同時に収めることができず、バッチ処理が必要な場合に備えてです。