1. 程式人生 > >深度學習框架tensorflow學習與應用3(非線性迴歸訓練示例)

深度學習框架tensorflow學習與應用3(非線性迴歸訓練示例)

在瞭解了迴歸演算法中的正向傳播和反向傳播之後, 我們可以用梯度下降法來進行一個非線性迴歸的示例.

此次示例中, 我們設定輸入樣本神經元只有一個, 中間神經元有10個, 輸出神經元1個.

                                       神經圖.jpg

這裡用到了matplotlib.pyplot庫, 用來畫出我們的訓練樣本資料影象和我們訓練得到的迴歸影象.

在看下面程式碼的時候注意, 如果你對迴歸演算法的學習開始於吳恩達先生的深度學習課程, 那麼你要注意下面程式碼中的權重W與樣本資料X的順序與轉置問題(相乘順序與行列轉置正好相反, 得到的結果是一樣的)


結果: