在具有固定效应的R中绘制OLS的预测()

时间:2016-05-03 15:29:45

标签: r lm predict

我想绘制给定回归的预测值,包括固定效果,但不绘制它们。在下面的例子中,我希望绘制y对y的简单回归,但考虑固定效应,使其不产生背景因子:

set.sed(123)            
df <- data.frame(x = rnorm(100), y = rbinom(100, 1, .5), f1 = sample(1:4, 100, T), f2 = sample(1:4, 100, T))        
mod <- lm(y ~ x + as.factor(f1) + as.factor(f2), data = df)         

new.x <- seq(min(df$x), max(df$x), length.out = 100)            
new.data <- data.frame(x = new.x, f1 = rep(mode(3), 100), f2 = mode(mean(2), 100))          
preds <- predict(mod, newdata = new.data, interval = 'confidence')          

plot(df$y ~ df$x + as.factor(df$f1) + as.factor(df$f2), type = 'n')         
polygon(c(rev(new.data$x), new.data$x), c(rev(preds[,3]), preds[,2]), col = "grey", border = NA)
abline(mod)
lines(new.data$x, preds[,3], lty = "dashed", col = "red")
lines(new.data$x, preds[,2], lty = "dashed", col = "red")

非常感谢任何和所有的想法。感谢。

0 个答案:

没有答案