套索回归是在Statsmodels中实现的吗？

Question

我喜欢在statsmodels中使用线性套索回归，因此能够使用“公式”符号来编写模型，这将节省我在处理许多分类变量及其交互时的大量编码时间。然而，它似乎还没有在统计模型中实现？

Answer 1

我们可以使用Patsy和scikit-learn来获得与statsmodels中的公式符号相同的结果。请参见下面的代码：

from patsy import dmatrices

# create dummy variables, and their interactions
y, X = dmatrices('outcome ~ C(var1)*C(var2)', df, return_type="dataframe")
# flatten y into a 1-D array so scikit-learn can understand it
y = np.ravel(y)

现在，我可以使用scikit learn中实现的任何模型，通常的符号是X作为自变量，y作为从属变量。