machine-learning - 预测期间的小批量大小

Question

我正在使用 lstm 模型。我了解小批量大小对于训练模型的意义。基本上它与批量更新梯度有关，而不是在每个样本之后更新。但是小批量大小在预测阶段意味着什么。我无法理解批量大小在预测阶段的作用。改变它会影响我的结果吗？

标签： machine-learninglstmpredictionupdatebatchsize

Answer 1

批量大小等仅与学习有关。在您的模型学习（训练）后，它只会保存权重。在测试或预测时，它只会使用保存的权重进行预测。

默认情况下，普通 LSTM 在批量大小后重置单元状态，但您可以更改它。您可以使其在一个时期后更新状态，甚至保持所有状态。