【发布时间】:2015-02-19 05:29:30
【问题描述】:
我有一个 8 列的 x 矩阵。我想运行glmnet 进行套索回归。我知道我需要打电话:
glmnet(x, y, family = "binomial", ...).
但是,我如何让x 也考虑所有单向交互?我是否必须手动重新制作数据框:如果是这样,有没有更简单的方法?我想我希望使用 R 公式做一些事情。
【问题讨论】:
标签: r formula interaction glmnet
我有一个 8 列的 x 矩阵。我想运行glmnet 进行套索回归。我知道我需要打电话:
glmnet(x, y, family = "binomial", ...).
但是,我如何让x 也考虑所有单向交互?我是否必须手动重新制作数据框:如果是这样,有没有更简单的方法?我想我希望使用 R 公式做一些事情。
【问题讨论】:
标签: r formula interaction glmnet
是的,有一种方便的方法。其中两个步骤很重要。
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)
【讨论】:
f <- as.formula( ~ .*.) 和x <- model.matrix(f, TrainData)[, -1],然后选择train(x = x, ...)。
f <- as.formula( ~ .^2) 也应该适用于包括主效应和所有成对的交互作用
【讨论】: