从列生成回归公式

Question

我正在尝试使用负二项式回归生成一个用于自动分析变量及其相互作用的模型。我已经取得了一些进展（主要归功于 SO :-)，现在我想生成回归字符串以传递给 neg bin 模型和 stepaic 函数。

有什么想法吗？

这是我的代码

library(dplyr)
library(purrr)
library(stringr)
df<- nycflights13::weather
interactions <- c("hour","dewp","humid","wind_dir","wind_speed")
target_col<- "temp"
out <- combn(interactions, 2, function(x) df %>%
               dplyr::select(x) %>%
               transmute(!! str_c(x, collapse="_"):= reduce(., `*`)),
             simplify = FALSE) %>% 
  bind_cols %>% 
  bind_cols(df, .)

target_formula<- target_col ~ interactions + combinations
model <- glm.nb(target_formula, data=df)
model<- stepaic(model)

Answer 1

我看到您有 5 个变量要在公式中使用。而且您还希望包含交互。

那你为什么不试试这个方法：

df <- nycflights13::weather
interactions <- c("hour","dewp","humid","wind_dir","wind_speed")
target_col <- "temp"
target_formula <- as.formula(sprintf("%s ~ (%s)^2", 
                             target_col, 
                             paste(interactions, collapse = " + ")))
model <- MASS::glm.nb(target_formula, data = df)
model <- MASS::stepAIC(model)

---

编辑

如果您只想计算某些变量的交互作用，您可以执行以下操作：

df<- nycflights13::weather
vars <- c("hour","dewp","humid","wind_dir","wind_speed", 
          "wind_dir:wind_speed", "dewp:humid")
target_col <- "temp"
target_formula <- as.formula(sprintf("%s ~ %s", 
                                     target_col, 
                                     paste(vars, collapse = " + ")))
model <- MASS::glm.nb(target_formula, data = df)
model <- MASS::stepAIC(model)

只需以var1:var2 之类的格式将它们添加到vars。

---

编辑 2

如果您需要所有只是一些变量之间的交互，您可以按照以下方式进行：

df<- nycflights13::weather
singles <- c("hour","dewp")
interactions <- c("humid","wind_dir","wind_speed")
target_col <- "temp"

target_formula <- as.formula(sprintf("%s ~  %s + (%s)^2", 
                                     target_col, 
                                     paste(singles, collapse = " + "),
                                     paste(interactions, collapse = " + ")))
model <- MASS::glm.nb(target_formula, data = df)
model <- MASS::stepAIC(model)

问题是你会得到所有的交互：例如，如果你不想要humid:wind_dir，这个解决方案不是你想要的。

如果你只想要特定的交互，你需要使用之前的编辑。

如果你想使用除了一些之外的大部分交互，还有其他的方式，但是如果你有特定的规则，我们可以想出一些高效的方式。

例如，像这样的：

setdiff(combn(c("humid","wind_dir","wind_speed"), 2, paste, collapse = ":"), "humid:wind_dir")
#> [1] "humid:wind_speed"    "wind_dir:wind_speed"

Answer 2

使用reformulate 创建没有交互的公式，然后使用update 它包含交互：

rhs_vars <- c("hour","dewp","humid","wind_dir","wind_speed")
target <- "temp"

update(reformulate(rhs_vars, target), . ~ (.)^2)

给出这个公式对象：

temp ~ hour + dewp + humid + wind_dir + wind_speed + hour:dewp + 
    hour:humid + hour:wind_dir + hour:wind_speed + dewp:humid + 
    dewp:wind_dir + dewp:wind_speed + humid:wind_dir + humid:wind_speed + 
    wind_dir:wind_speed