8000 TST make TheilSenRegressor run faster in common tests, I could swear … · scikit-learn/scikit-learn@97d8341 · GitHub
[go: up one dir, main page]

Skip to content

Commit 97d8341

Browse files
committed
TST make TheilSenRegressor run faster in common tests, I could swear I did that before...
1 parent 8cf714c commit 97d8341

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

sklearn/utils/estimator_checks.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,9 @@ def set_fast_parameters(estimator):
8888
# be tolerant of noisy datasets (not actually speed)
8989
estimator.set_params(alpha=.5)
9090

91+
if estimator.__class__.__name__ == "TheilSenRegressor":
92+
estimator.max_subpopulation = 100
93+
9194
if isinstance(estimator, BaseRandomProjection):
9295
# Due to the jl lemma and often very few samples, the number
9396
# of components of the random matrix projection will be probably

0 commit comments

Comments
 (0)
0