Skip to content

Commit bd4fd12

Browse files
committed
BUG fix pruning doc
1 parent 80f08cd commit bd4fd12

File tree

2 files changed

+35
-36
lines changed

2 files changed

+35
-36
lines changed

examples/plot_pruning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def _fastcan_pruning(
7272
).fit(X)
7373
atoms = kmeans.cluster_centers_
7474
ids_fastcan = minibatch(
75-
X.T, atoms.T, n_samples_to_select, batch_size=batch_size, tol=1e-6, verbose=0
75+
X.T, atoms.T, n_samples_to_select, batch_size=batch_size, tol=1e-9, verbose=0
7676
)
7777
pruned_lr = LogisticRegression(max_iter=110).fit(X[ids_fastcan], y[ids_fastcan])
7878
return pruned_lr.coef_, pruned_lr.intercept_

pixi.lock

Lines changed: 34 additions & 35 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)