如何在使用glmnet之前进行所有交互

时间:2014-12-19 23:58:24

标签: r formula interaction glmnet

我有一个8列的x矩阵。我想运行glmnet来进行套索回归。我知道我需要打电话:

glmnet(x, y, family = "binomial", ...). 

但是,如何让x考虑所有单向交互呢?我是否必须手动重新制作数据框:如果是,是否有更简单的方法?我想我希望用R公式做点什么。

2 个答案:

答案 0 :(得分:29)

是的,有一个方便的方法。其中的两个步骤很重要。

library(glmnet)
# Sample data
data <- data.frame(matrix(rnorm(9 * 10), ncol = 9))
names(data) <- c(paste0("x", 1:8), "y")
# First step: using .*. for all interactions
f <- as.formula(y ~ .*.)
y <- data$y
# Second step: using model.matrix to take advantage of f
x <- model.matrix(f, data)[, -1]
glmnet(x, y)

答案 1 :(得分:2)

f <- as.formula( ~ .^2)也应该用于包含主效应和所有成对交互