我正在尝试在自定义的r函数中使用r基逻辑回归函数,但是glm()
无法识别我的变量。我已经在搜索引擎中尝试了多个搜索关键字,但是所有答案都与拟合逻辑回归相关。
例如:
第一次尝试:
dat <- data.frame(a = c(3,4,5), b = c("a","a","b"))
logit <- function(dataname, x, y) {
model = glm(y ~ x, data = dataname, family = "binomial")
model
}
logit(dat, a, b)
Error in eval(predvars, data, env) : object 'b' not found
替代方法:
logit <- function(dataname, x, y) {
model = glm(eval(substitute(y), dat) ~ eval(substitute(x), dat), family = "binomial")
model
}
logit(dat, a, b)
输出将把我的IV更改为eval(substitute(x),数据名)而不是x。
glm.fit: fitted probabilities numerically 0 or 1 occurred
Call: glm(formula = eval(substitute(y), dataname) ~ eval(substitute(x),
dataname), family = "binomial")
Coefficients:
(Intercept) eval(substitute(x), dataname)
-208.27 46.34
Degrees of Freedom: 2 Total (i.e. Null); 1 Residual
Null Deviance: 3.819
Residual Deviance: 3.597e-10 AIC: 4
有什么办法可以在输出中以正确的IV名称获得正确的输出?
谢谢
我同意@IceCreamToucan的最佳方法是将公式传递给函数
logit <- function(dataname, formula) {
model = glm(formula, data = dataname, family = "binomial")
model
}
logit(dat, b~a)
否则,您应该首先构建公式,然后将其传递给glm
logit <- function(dataname, x, y) {
formula <- reformulate(as.character(substitute(x)), as.character(substitute(y)))
model = glm(formula, data = dataname, family = "binomial")
model
}
logit(dat, a, b)
logit <- function(dataname, x, y) {
model = glm( as.formula(paste(y, "~", x)),
data = dataname,
family = "binomial")
model
}
logit(dat, "a", "b")
如果您有许多解释变量,则可以传入名称向量并使用:
as.formula(paste(y, "~", paste(x, collapse="+")))