ASV Fix tol in SGDRegressorBenchmark (#26146)

OmarManzoor · jeremiedbb · web-flow · commit fb02c174c0f8 · 2023-04-14T10:24:17.000Z
Co-authored-by: jeremie du boisberranger &lt;jeremiedbb@yahoo.fr&gt;
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
 
