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:
      author  = {Ismael Lemhadri and Feng Ruan and Louis Abraham and Robert Tibshirani},
      title   = {LassoNet: A Neural Network with Feature Sparsity},
      journal = {Journal of Machine Learning Research},
      year    = {2021},
      volume  = {22},
      number  = {127},
      pages   = {1-29},
      url     = {}