LassoNet: Neural Networks with Feature Sparsity

Ismael Lemhadri Feng Ruan Louis Abraham Rob Tibshirani

LassoNet is a method for feature selection in neural networks, to enhance interpretability of the final network.

LassoNet in 2 minutes


pip install lassonet


LassoNet sometimes require fine tuning. For optimal performance, consider:

Intro video



The algorithms and method used in this package came primarily out of research in Rob Tibshirani's lab at Stanford University. If you use LassoNet in your research we would appreciate a citation to the paper:
        title={LassoNet: Neural Networks with Feature Sparsity},
        author={Lemhadri, Ismael and Ruan, Feng and
                Abraham, Louis and Tibshirani, Robert},
        journal={arXiv preprint arXiv:1907.12207},