在模拟多元数据进行回归时,如何设置R平方(包括示例代码)？

Question

我正在尝试模拟三变量数据集,以便我可以在其上运行线性回归模型.'X1'和'X2'将是连续的独立变量(mean = 0,sd = 1),'Y'将是连续因变量.

变量将是回归模型将产生如下系数:Y = 5 + 3(X1) - 2(X2)

我想模拟这个数据集,使得得到的回归模型的R平方值为0.2.如何确定'sd.value'的值,以便回归模型具有此R平方？

n <- 200 
set.seed(101) 
sd.value <- 1

X1 <- rnorm(n, 0, 1)
X2 <- rnorm(n, 0, 1)
Y <- rnorm(n, (5 + 3*X1 - 2*X2), sd.value)

simdata <- data.frame(X1, X2, Y)

summary(lm(Y ~ X1 + X2, data=simdata))

Answer 1

看看这段代码,它应该足够接近你想要的东西:

simulate <- function(n.obs=10^4, beta=c(5, 3, -2), R.sq=0.8) {
    stopifnot(length(beta) == 3)
    df <- data.frame(x1=rnorm(n.obs), x2=rnorm(n.obs))  # x1 and x2 are independent
    var.epsilon <- (beta[2]^2 + beta[3]^2) * (1 - R.sq) / R.sq
    stopifnot(var.epsilon > 0)
    df$epsilon <- rnorm(n.obs, sd=sqrt(var.epsilon))
    df$y <- with(df, beta[1] + beta[2]*x1 + beta[3]*x2 + epsilon)
    return(df)
}
get.R.sq <- function(desired) {
    model <- lm(y ~ x1 + x2, data=simulate(R.sq=desired))
    return(summary(model)$r.squared)
}
df <- data.frame(desired.R.sq=seq(from=0.05, to=0.95, by=0.05))
df$actual.R.sq <- sapply(df$desired.R.sq, FUN=get.R.sq)
plot(df)
abline(a=0, b=1, col="red", lty=2)

基本上你的问题归结为找出var.epsilon的表达式.由于我们有y = b1 + b2*x1 + b3*x2 + epsilon,而Xs和epsilon都是独立的,我们有var [y] = b2 ^ 2*var [x1] + b3 ^ 2*var [x2] + var [eps],其中var [Xs] = 1假设.然后,您可以求解var [eps]作为R平方的函数.