ASV Fix tol in SGDRegressorBenchmark (scikit-learn#26146)

Co-authored-by: jeremie du boisberranger <[email protected]>
engkenni · Apr 14, 2023 · fb02c17 · fb02c17
1 parent 22ca6ff
commit fb02c17
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/asv_benchmarks/benchmarks/linear_model.py b/asv_benchmarks/benchmarks/linear_model.py
@@ -164,7 +164,11 @@ def make_data(self, params):
         return data
 
     def make_estimator(self, params):
-        estimator = SGDRegressor(max_iter=1000, tol=1e-16, random_state=0)
+        (representation,) = params
+
+        max_iter = 60 if representation == "dense" else 300
+
+        estimator = SGDRegressor(max_iter=max_iter, tol=None, random_state=0)
 
         return estimator