web-dev-qa-db-ja.com

Kerasで予測にバッチサイズが必要なのはなぜですか?

Kerasでは、データテストのクラスを予測するために、predict_classes()が使用されます。

例えば:

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

私の質問は、batch_sizeトレーニング中ですが、なぜbatch_size予測のために?どのように機能しますか?

31
malioboro

Kerasは複数の値を同時に予測できます。たとえば、100要素のベクトルを入力した場合、Kerasは各要素に対して1つの予測を計算し、100の出力を生成できます。この計算は、batch_sizeで定義されたバッチで実行することもできます。

これは、CPU/GPU RAMにすべてのデータを同時に収めることができず、バッチ処理が必要な場合に備えてです。

30