lasso-net / lassonet

Feature selection in neural networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add documentation for LassoNetAutoEncoder

ElrondL opened this issue · comments

In the example mnist_ae.py the module LassoNetAutoEncoder is imported from lassonet, but this is not in the documentation?

Does LassoNet have an autoencoder implementation? If not, is it possible to create one by connecting two LassoNetRegressors?

@ilemhadri do you know how the autoencoders were implemented in the paper?

My best guess is that it was something along the lines of

from sklearn.datasets import fetch_california_housing
from sklearn.preprocessing import StandardScaler

from lassonet import LassoNetRegressor

X, _ = fetch_california_housing(return_X_y=True)
X = StandardScaler().fit_transform(X)

model = LassoNetRegressor(verbose=2)
path = model.path(X, X)

My reference is #25