diff --git a/02 Linear Regression/LinearRegression.py b/02 Linear Regression/LinearRegression.py index 8b8edba..842b187 100644 --- a/02 Linear Regression/LinearRegression.py +++ b/02 Linear Regression/LinearRegression.py @@ -17,8 +17,8 @@ def fit(self, X, y): for _ in range(self.n_iters): y_pred = np.dot(X, self.weights) + self.bias - dw = (1/n_samples) * np.dot(X.T, (y_pred-y)) - db = (1/n_samples) * np.sum(y_pred-y) + dw = (1/n_samples) * 2*np.dot(X.T, (y_pred-y)) + db = (1/n_samples) * 2*np.sum(y_pred-y) self.weights = self.weights - self.lr * dw self.bias = self.bias - self.lr * db