mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2024-11-27 15:01:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor checks

Browse Source
This commit is contained in:
jbsch 2024-10-24 12:37:54 +05:30
parent 254b9bf87e
commit 97eb853842
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
machine_learning/ridge_regression/test_ridge_regression.py
View File

@ -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])