File tree Expand file tree Collapse file tree 2 files changed +7
-5
lines changed Expand file tree Collapse file tree 2 files changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -33,9 +33,10 @@ class BayesianRidge(RegressorMixin, LinearModel):
33
33
34
34
Parameters
35
35
----------
36
- max_iter : int, default=300
36
+ max_iter : int, default=None
37
37
Maximum number of iterations over the complete dataset before
38
- stopping independently of any early stopping criterion.
38
+ stopping independently of any early stopping criterion. If `None`, it
39
+ corresponds to `max_iter=300`.
39
40
40
41
.. versionchanged:: 1.3
41
42
Original file line number Diff line number Diff line change @@ -294,9 +294,9 @@ def test_dtype_correctness(Estimator):
294
294
np .testing .assert_allclose (coef_32 , coef_64 , rtol = 1e-4 )
295
295
296
296
297
+ # TODO(1.5) remove
297
298
def test_bayesian_ridge_n_iter_deprecated ():
298
- # check the deprecation warning of n_iter
299
- # TODO(1.5) remove
299
+ """Check the deprecation warning of `n_iter`."""
300
300
depr_msg = (
301
301
"'n_iter' was renamed to 'max_iter' in version 1.3 and will be removed in 1.5"
302
302
)
@@ -307,8 +307,9 @@ def test_bayesian_ridge_n_iter_deprecated():
307
307
est .fit (X , y )
308
308
309
309
310
+ # TODO(1.5) remove
310
311
def test_bayesian_ridge_max_iter_and_n_iter_both_set ():
311
- # TODO(1.5) remove
312
+ """Check that we raise an error if both `max_iter` and `n_iter` are set."""
312
313
"""Check that a ValueError is raised when both `max_iter` and `n_iter` are set."""
313
314
err_msg = (
314
315
"Both `n_iter` and `max_iter` attributes were set. Attribute"
You can’t perform that action at this time.
0 commit comments