neural-network - 为什么在 Keras 中预测需要批量大小？

Question

在 Keras 中，为了预测数据测试的类别，predict_classes()用来。

例如:

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

我的问题是，我知道 batch_size 的用法在训练中，但为什么需要 batch_size为了预测？它是如何工作的？

Answer 1

Keras 可以同时预测多个值，就像你输入一个包含 100 个元素的向量一样，Keras 可以为每个元素计算一个预测，给出 100 个输出。此计算也可以分批完成，由 batch_size 定义。

这是以防万一您无法同时将所有数据放入 CPU/GPU RAM 中，并且需要进行批处理。