一 介绍
下面介绍在TensorFlow中进行随机梯度下降优化的函数。
在TensorFlow中通过一个叫Optimizer的优化器类进行训练优化。
二 梯度下降优化器
三 说明
在训练过程中先实例化一个优化函数如tf.train.GradientDescentOptimizer,并基于一定的学习率进行梯度优化训练。
optimizer = tf.train.GradientDescentOptimizer(learning_rate)
接着使用一个minimize()操作,里面传入损失值节点loss,再启动一个外层的循环,优化器就会按照循环的次数一次次沿着loss最小值的方向优化参数了。
整个过程中的求导和反向传播操作,都是在优化器里自动完成的。
目前比较常用的优化器为Adam优化器。
四 参考