From c3b0a21ddf7e97b93adbb129caaa2a9b533bd32e Mon Sep 17 00:00:00 2001 From: Guido Gallopyn Date: Tue, 26 Oct 2021 08:06:04 -0400 Subject: [PATCH] Update 05_1_gradientdescent_manually.py bug fix in gradient calculation --- 05_1_gradientdescent_manually.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/05_1_gradientdescent_manually.py b/05_1_gradientdescent_manually.py index 833a48d7..1a79f03f 100644 --- a/05_1_gradientdescent_manually.py +++ b/05_1_gradientdescent_manually.py @@ -22,7 +22,7 @@ def loss(y, y_pred): # J = MSE = 1/N * (w*x - y)**2 # dJ/dw = 1/N * 2x(w*x - y) def gradient(x, y, y_pred): - return np.dot(2*x, y_pred - y).mean() + return np.mean(2*x*(y_pred - y)) print(f'Prediction before training: f(5) = {forward(5):.3f}')