python - python 中 "Lasso"的输入可以包含分类变量吗？

Question

我想用 lasso 在 python 中执行多元线性回归。我不确定输入观察矩阵 X 是否可以包含分类变量。我从这里阅读了说明:lasso in python

但它很简单，并没有指明允许的类型。例如，我的代码包括:

model = Lasso(fit_intercept=False, alpha=0.01)
model.fit(X, y)

在上面的代码中，X 是一个大小为 n×p 的观察矩阵，p 个变量中的一个可以是分类类型吗？

Answer 1

您需要使用 1 和 0 表示分类变量。如果您的分类变量是二元的，这意味着每个属于两个类别之一，那么您将所有类别 A 和 B 变量分别替换为 0 和 1。如果有些有两个以上的类别，您将需要使用 dummy variables .

我通常将数据放在 Pandas 数据框中，在这种情况下，我使用 houses = pd.get_dummies(houses)，它会创建虚拟变量。