Skip to content

Commit cc18036

Browse files
Merge pull request patrickloeber#10 from guidogallopyn/patch-1
Update 05_1_gradientdescent_manually.py
2 parents 75d768b + c3b0a21 commit cc18036

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

05_1_gradientdescent_manually.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def loss(y, y_pred):
2222
# J = MSE = 1/N * (w*x - y)**2
2323
# dJ/dw = 1/N * 2x(w*x - y)
2424
def gradient(x, y, y_pred):
25-
return np.dot(2*x, y_pred - y).mean()
25+
return np.mean(2*x*(y_pred - y))
2626

2727
print(f'Prediction before training: f(5) = {forward(5):.3f}')
2828

0 commit comments

Comments
 (0)