diff --git a/examples/plot_pruning.py b/examples/plot_pruning.py index d1d3c69..08b8ba1 100644 --- a/examples/plot_pruning.py +++ b/examples/plot_pruning.py @@ -75,6 +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]) return pruned_lr.coef_, pruned_lr.intercept_