From 97eb85384295dcadd71ea255dcfc4db9382471f4 Mon Sep 17 00:00:00 2001 From: jbsch Date: Thu, 24 Oct 2024 12:37:54 +0530 Subject: [PATCH] minor checks --- .../ridge_regression/test_ridge_regression.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/machine_learning/ridge_regression/test_ridge_regression.py b/machine_learning/ridge_regression/test_ridge_regression.py index 1153a1d85..71d485bfa 100644 --- a/machine_learning/ridge_regression/test_ridge_regression.py +++ b/machine_learning/ridge_regression/test_ridge_regression.py @@ -11,7 +11,8 @@ To run these tests, use the following command: python -m doctest test_ridge_regression.py -v """ -# from machine_learning.ridge_regression import RidgeRegression +import numpy as np # noqa: F401 +from ridge_regression import RidgeRegression # noqa: F401 def test_feature_scaling(): @@ -37,8 +38,8 @@ def test_fit(): Tests the fit function of RidgeRegression -------- >>> model = RidgeRegression(alpha=0.01, - regularization_param=0.1, - num_iterations=1000) + ... regularization_param=0.1, + ... num_iterations=1000) >>> X = np.array([[1], [2], [3]]) >>> y = np.array([2, 3, 4]) @@ -59,8 +60,8 @@ def test_predict(): Tests the predict function of RidgeRegression -------- >>> model = RidgeRegression(alpha=0.01, - regularization_param=0.1, - num_iterations=1000) + ... regularization_param=0.1, + ... num_iterations=1000) >>> X = np.array([[1], [2], [3]]) >>> y = np.array([2, 3, 4])