Skip to content

Commit cc4ea72

Browse files
mrleuAlex Leu
authored andcommitted
updated epoch and training size (scikit-learn#21773)
Co-authored-by: Alex Leu <[email protected]>
1 parent ec50455 commit cc4ea72

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/linear_model/plot_sparse_logistic_regression_20newsgroups.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
solver = "saga"
4141

4242
# Turn down for faster run time
43-
n_samples = 10000
43+
n_samples = 5000
4444

4545
X, y = fetch_20newsgroups_vectorized(subset="all", return_X_y=True)
4646
X = X[:n_samples]
@@ -58,8 +58,8 @@
5858
)
5959

6060
models = {
61-
"ovr": {"name": "One versus Rest", "iters": [1, 2, 4]},
62-
"multinomial": {"name": "Multinomial", "iters": [1, 3, 7]},
61+
"ovr": {"name": "One versus Rest", "iters": [1, 2, 3]},
62+
"multinomial": {"name": "Multinomial", "iters": [1, 2, 5]},
6363
}
6464

6565
for model in models:

0 commit comments

Comments
 (0)