跳转到主要内容

机器学习中Batch Size、Iteration和Epoch的概念

demi 提交于

<strong>Batch Size:</strong>批尺寸。

机器学习中参数更新的方法有三种:

(1)Batch Gradient Descent,批梯度下降,遍历全部数据集计算一次损失函数,进行一次参数更新,这样得到的方向能够更加准确的指向极值的方向,但是计算开销大,速度慢;

(2)Stochastic Gradient Descent,随机梯度下降,对每一个样本计算一次损失函数,进行一次参数更新,优点是速度快,缺点是方向波动大,忽东忽西,不能准确的指向极值的方向,有时甚至两次更新相互抵消;

(3)Mini-batch Gradient Decent,小批梯度下降,前面两种方法的折中,把样本数据分为若干批,分批来计算损失函数和更新参数,这样方向比较稳定,计算开销也相对较小。Batch Size就是每一批的样本数量。

<strong>Iteration:</strong>迭代,可以理解为w和b的一次更新,就是一次Iteration。

<strong>Epoch:</strong>样本中的所有样本数据被计算一次就叫做一个Epoch。

本文转自:博客园 - <a href="https://www.cnblogs.com/mstk/p/8214499.html">代码空间</a&gt;,转载此文目的在于传递更多信息,版权归原作者所有。