多元线性回归：用户定义函数错误

Question

我已经为 MLR 编写了函数。但是，输出似乎有问题（请参阅最后的示例）。

但是当我运行代码时，逐行输出是正确的。

mlr <- function(dependentvar, dataset) {

x <- model.matrix(dependentvar ~., dataset) # Design Matrix for x

y <- dependentvar # dependent variable

betas <- solve(crossprod(x))%*%crossprod(x,y) # beta values

SST <- t(y)%*%y - (sum(y)^2/dim(dataset)[1]) # total sum of squares

SSres <- t(y)%*%y -(t(betas)%*%crossprod(x,y))  # sum of squares of residuals

SSreg <- SST - SSres  # regression sum of squares

sigmasqr <- SSres/(length(y) - dim(dataset)[2])  # variance or (MSE)

varofbeta <- sigmasqr[1]*solve( crossprod(x)) # variance of beta

cat("SST:", SST,"SSresiduals:", SSres,"SSregression:", SSreg, sep = "\n", append = FALSE)

return(betas)

}

要查看问题，请尝试

mlr(trees$Height, trees)

即使我去掉$

我也遇到同样的问题

Height <- trees$Height
mlr(Height, trees)

Answer 1

使用以下内容：

x <- model.matrix(reformulate(".", dependentvar), dataset)
y <- dataset[[dependentvar]]

并将 dependentvar 作为字符串传入。

示例：

mlr("Height", trees)