Fix the issue #9.

trekhleb · Jan 10, 2019 · 04ecea6 · 04ecea6
1 parent d16d88e
commit 04ecea6
Show file tree

Hide file tree

Showing 4 changed files with 1,161 additions and 1,161 deletions.
diff --git a/homemade/linear_regression/linear_regression.py b/homemade/linear_regression/linear_regression.py
@@ -102,7 +102,7 @@ def gradient_step(self, alpha, lambda_param):
         # Vectorized version of gradient descent.
         theta = theta * reg_param - alpha * (1 / num_examples) * (delta.T @ self.data).T
         # We should NOT regularize the parameter theta_zero.
-        theta[1] = theta[1] - alpha * (1 / num_examples) * (self.data[:, 0].T @ delta).T
+        theta[0] = theta[0] - alpha * (1 / num_examples) * (self.data[:, 0].T @ delta).T
 
         self.theta = theta