Skip to content

Commit 97eb853

Browse files
committed
minor checks
1 parent 254b9bf commit 97eb853

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

Diff for: machine_learning/ridge_regression/test_ridge_regression.py

+6-5
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@
1111
python -m doctest test_ridge_regression.py -v
1212
"""
1313

14-
# from machine_learning.ridge_regression import RidgeRegression
14+
import numpy as np # noqa: F401
15+
from ridge_regression import RidgeRegression # noqa: F401
1516

1617

1718
def test_feature_scaling():
@@ -37,8 +38,8 @@ def test_fit():
3738
Tests the fit function of RidgeRegression
3839
--------
3940
>>> model = RidgeRegression(alpha=0.01,
40-
regularization_param=0.1,
41-
num_iterations=1000)
41+
... regularization_param=0.1,
42+
... num_iterations=1000)
4243
>>> X = np.array([[1], [2], [3]])
4344
>>> y = np.array([2, 3, 4])
4445
@@ -59,8 +60,8 @@ def test_predict():
5960
Tests the predict function of RidgeRegression
6061
--------
6162
>>> model = RidgeRegression(alpha=0.01,
62-
regularization_param=0.1,
63-
num_iterations=1000)
63+
... regularization_param=0.1,
64+
... num_iterations=1000)
6465
>>> X = np.array([[1], [2], [3]])
6566
>>> y = np.array([2, 3, 4])
6667

0 commit comments

Comments
 (0)