1. 程式人生 > >TensorFlow之三 ----- 機器學習初學者的MNIST

TensorFlow之三 ----- 機器學習初學者的MNIST

現在我們知道我們想要我們的模型來做什麼了,使用TensorFlow訓練它來做這些將會非常簡單。因為TensorFlow知道你的計算的整個圖表,它能自動使用逆傳演算法backpropagation algorithm來高效的決定你的變數如何影響你要求它最小化的損失。然後它能應用你的最佳演算法的選擇來修改變數並且減少損失。
train_step = tf.train.GradientDescentOptimizer(0.5).minimize(cross_entropy)
在這種情況下,我們要求TensorFlow使用帶有一個0.5的學習率的梯度下降演算法gradient descent algorithm來最小化交叉熵cross_entropy。梯度下降是一個簡單的過程,其中TensorFlow簡單的移動一點點每個變數在減少損失的方向。但是TensorFlow同樣提供了許多其他的優化演算法:使用一個就像調整一行一樣簡單。