1. 程式人生 > >TensorFlow中梯度下降函式

TensorFlow中梯度下降函式

一 介紹下面介紹在TensorFlow中進行隨機梯度下降優化的函式。在TensorFlow中通過一個叫Optimizer的優化器類進行訓練優化。二 梯度下降優化器三 說明在訓練過程中先例項化一個優化函式如tf.train.GradientDescentOptimizer,並基於一定的學習率進行梯度優化訓練。optimizer = tf.train.GradientDescentOptimizer(learning_rate)接著使用一個minimize()操作,裡面傳入損失值節點loss,再啟動一個外層的迴圈,優化器就會按照迴圈的次數一次次沿著loss最小值的方向優化引數了。整個過程中的求導和反向傳播操作,都是在優化器裡自動完成的。目前比較常用的優化器為Adam優化器。四 參考