Reference implementation of LassoNet
Project description
LassoNet
This project is about performing feature selection in neural networks. At the moment, we support fully connected feed-forward neural networks. LassoNet is based on the work presented in this paper (bibtex here for citation). Here is a link to the promo video:
Code
We have designed the code to follow scikit-learn's standards to the extent possible (e.g. linear_model.Lasso).
To install it,
pip install lassonet
Our plan is to add more functionality that help users understand the important features in neural networks.
Usage
from lassonet import LassoNetClassifierCV
model = LassoNetClassifierCV() # LassoNetRegressorCV
path = model.fit(X_train, y_train)
print("Best model scored", model.score(X_test, y_test))
print("Lambda =", model.best_lambda_)
Website
LassoNet's website is https://lassonet.ml. It contains many useful references including the paper, live talks and additional documentation.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for lassonet-0.0.10-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 564bf5d684804a36cea05ab9e15c63d6e59a2159baf962b69fbc47789573e0a3 |
|
MD5 | a524336b08194ed671624525c58516ef |
|
BLAKE2b-256 | 4e037e3bb6bfd401f8e3262e4c4057713b2b14a9382fd7dec477149e56c58272 |