資源簡介
程序利用python numpy和matplotlib庫實現了簡單的梯度下降算法并對其進行可視化。程序通過簡單的可視化過程解釋了梯度下降的原理,供大家學習參考。注:并不是反向傳播的實現,而是對梯度下降原理的解釋,請根據需要自行參考。
代碼片段和文件信息
import?numpy?as?np
import?matplotlib.pyplot?as?plt
x?=?np.linspace(-16141)
y?=?(x-2.5)**2-1
def?dJ(theta):
????return?2*(theta-2.5)
def?J(theta):
????return?(theta-2.5)**2-1
theta?=?0.0
eta?=?0.1
epsilon?=?1e-8
theta_history?=?[]
while?True:
????gradient?=?dJ(theta)?
????l
評論
共有 條評論