Merge pull request #46 from MatthewSZhang/minibatch-doc

MatthewSZhang · web-flow · commit 392f0284c952 · 2025-02-21T13:47:49.000+08:00
DOC debug end of ids
diff --git a/examples/plot_pruning.py b/examples/plot_pruning.py
@@ -75,7 +75,7 @@ def _fastcan_pruning(
         X.T, atoms.T, n_samples_to_select, batch_size=batch_size, tol=1e-9, verbose=0
     )
     pruned_lr = LogisticRegression(max_iter=110).fit(X[ids_fastcan], y[ids_fastcan])
-    print(atoms[0], ids_fastcan[:10])
+    print(atoms[-1], ids_fastcan[-10:])
     return pruned_lr.coef_, pruned_lr.intercept_
 
 

Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,7 @@ def _fastcan_pruning(`
`75`	`75`	`X.T, atoms.T, n_samples_to_select, batch_size=batch_size, tol=1e-9, verbose=0`
`76`	`76`	`)`
`77`	`77`	`pruned_lr = LogisticRegression(max_iter=110).fit(X[ids_fastcan], y[ids_fastcan])`
`78`		`- print(atoms[0], ids_fastcan[:10])`
	`78`	`+ print(atoms[-1], ids_fastcan[-10:])`
`79`	`79`	`return pruned_lr.coef_, pruned_lr.intercept_`
`80`	`80`
`81`	`81`