Is LASSO regression implemented in Statsmodels?

python regression statsmodels lasso-regression
famargar picture famargar · Apr 17, 2017 · Viewed 8.7k times · Source

I would love to use a linear LASSO regression within statsmodels, so to be able to use the 'formula' notation for writing the model, that would save me quite some coding time when working with many categorical variables, and their interactions. However, it seems like it is not implemented yet in stats models?

Answer

DiveIntoML picture DiveIntoML · Jun 16, 2017

Lasso is indeed implemented in statsmodels. The documentation is given in the url below:

http://www.statsmodels.org/dev/generated/statsmodels.regression.linear_model.OLS.fit_regularized.html

To be precise, the implementation in statsmodel has both L1 and L2 regularization, with their relative weight indicated by L1_wt parameter. You should look at the formula at the bottom to make sure you are doing exactly what you want to do.

Besides the elastic net implementation, there is also a square root Lasso method implemented in statsmodels.

Related questions

Run an OLS regression with Pandas Data Frame

I have a pandas data frame and I would like to able to predict the values of column A from the values in columns B and C. Here is a toy example: import pandas as pd df = pd.DataFrame({"A": [10,20,30,40,50], "…

Read More
Fixed effect in Pandas or Statsmodels

Is there an existing function to estimate fixed effect (one-way or two-way) from Pandas or Statsmodels. There used to be a function in Statsmodels but it seems discontinued. And in Pandas, there is something called plm, but I can't import …

Read More
ValueError: endog must be in the unit interval

While using statsmodels, I am getting this weird error: ValueError: endog must be in the unit interval. Can someone give me more information on this error? Google is not helping. Code that produced the error: """ Multiple regression with dummy variables. """ …

Read More