diff --git a/examples/plot_pruning.py b/examples/plot_pruning.py index 08b8ba1..b9d3fde 100644 --- 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_