Updated code as per PR feedback 6

Author: prajwal-38

machine_learning/ridge_regression.py

@@ -3,9 +3,12 @@
 from matplotlib import pyplot as plt
 from sklearn import datasets
 
+
 # Ridge Regression function
 # reference : https://en.wikipedia.org/wiki/Ridge_regression
-def ridge_cost_function(x: np.ndarray, y: np.ndarray, theta: np.ndarray, alpha: float) -> float:
+def ridge_cost_function(
+    x: np.ndarray, y: np.ndarray, theta: np.ndarray, alpha: float
+) -> float:
     """
     Compute the Ridge regression cost function with L2 regularization.
 
@@ -27,12 +30,26 @@ def ridge_cost_function(x: np.ndarray, y: np.ndarray, theta: np.ndarray, alpha:
     """
     m = len(y)
     predictions = np.dot(x, theta)
+<<<<<<< HEAD
     cost = (1 / (2 * m)) * np.sum((predictions - y) ** 2) + \
         (alpha / 2) * np.sum(theta[1:] ** 2)
+=======
+    cost = (1 / (2 * m)) * np.sum((predictions - y) ** 2) + (alpha / 2) * np.sum(
+        theta[1:] ** 2
+    )
+>>>>>>> c8c1d9a5896ed6f64a71a2f9392eb4ecc7ffff12
 
     return cost
 
-def ridge_gradient_descent(x: np.ndarray, y: np.ndarray, theta: np.ndarray, alpha: float, learning_rate: float, max_iterations: int) -> np.ndarray:
+
+def ridge_gradient_descent(
+    x: np.ndarray,
+    y: np.ndarray,
+    theta: np.ndarray,
+    alpha: float,
+    learning_rate: float,
+    max_iterations: int,
+) -> np.ndarray:
     """
     Perform gradient descent to minimize the cost function and fit the Ridge regression model.
 
@@ -62,8 +79,13 @@ def ridge_gradient_descent(x: np.ndarray, y: np.ndarray, theta: np.ndarray, alph
 
     return theta
 
+<<<<<<< HEAD
+=======
+
+>>>>>>> c8c1d9a5896ed6f64a71a2f9392eb4ecc7ffff12
 if __name__ == "__main__":
     import doctest
+
     doctest.testmod()
 
     # Load California Housing dataset
@@ -83,7 +105,9 @@ def ridge_gradient_descent(x: np.ndarray, y: np.ndarray, theta: np.ndarray, alph
     learning_rate = 0.01
     max_iterations = 1000
 
-    optimized_theta = ridge_gradient_descent(x, y, theta_initial, alpha, learning_rate, max_iterations)
+    optimized_theta = ridge_gradient_descent(
+        x, y, theta_initial, alpha, learning_rate, max_iterations
+    )
     print(f"Optimized theta: {optimized_theta}")
 
     # Prediction
@@ -94,8 +118,8 @@ def predict(x, theta):
 
     # Plotting the results (here we visualize predicted vs actual values)
     plt.figure(figsize=(10, 6))
-    plt.scatter(y, y_pred, color='b', label='Predictions vs Actual')
-    plt.plot([min(y), max(y)], [min(y), max(y)], color='r', label='Perfect Fit')
+    plt.scatter(y, y_pred, color="b", label="Predictions vs Actual")
+    plt.plot([min(y), max(y)], [min(y), max(y)], color="r", label="Perfect Fit")
     plt.xlabel("Actual values")
     plt.ylabel("Predicted values")
     plt.title("Ridge Regression: Actual vs Predicted Values")