updated epoch and training size (#21773)

mrleu · Alex Leu · glemaitre · commit 51a43ff6831b · 2021-12-25T12:26:49.000+01:00
Co-authored-by: Alex Leu &lt;alexcleu@ucla.edu&gt;
diff --git a/examples/linear_model/plot_sparse_logistic_regression_20newsgroups.py b/examples/linear_model/plot_sparse_logistic_regression_20newsgroups.py
@@ -40,7 +40,7 @@
 solver = "saga"
 
 # Turn down for faster run time
-n_samples = 10000
+n_samples = 5000
 
 X, y = fetch_20newsgroups_vectorized(subset="all", return_X_y=True)
 X = X[:n_samples]
@@ -58,8 +58,8 @@
 )
 
 models = {
-    "ovr": {"name": "One versus Rest", "iters": [1, 2, 4]},
-    "multinomial": {"name": "Multinomial", "iters": [1, 3, 7]},
+    "ovr": {"name": "One versus Rest", "iters": [1, 2, 3]},
+    "multinomial": {"name": "Multinomial", "iters": [1, 2, 5]},
 }
 
 for model in models:
