当回归有一个因素时 R 中的预测值

Question

我有以下回归：

a <- lm(y ~ factor(x) + z + factor(x) * z, data = dataset)

我想获得 x = 1 时的预测值，以及不同水平的 z。我一直在努力使用 predict 包来做到这一点。

如有任何帮助，我们将不胜感激。

Answer 1

对于以后的帖子，始终包含示例数据是一种很好的做法。请参阅 here how to provide a minimal reproducible example/attempt，包括示例数据。

除此之外，这是一个基于我生成的一些示例数据的简单示例。

# Generate sample data
set.seed(2017);
x <- as.numeric(gl(2, 10, 20));
z <- 1:20;
y <- 4 * x + 0.5 * z + rnorm(20);

# Fit model
fit <- lm(y ~ as.factor(x) + z + as.factor(x) * z);
summary(fit);
#    
#Call:
#lm(formula = y ~ as.factor(x) + z + as.factor(x) * z)
#
#Residuals:
#    Min      1Q  Median      3Q     Max
#-1.9283 -0.4702 -0.1270  0.7932  1.6648
#
#Coefficients:
#                Estimate Std. Error t value Pr(>|t|)
#(Intercept)      4.13695    0.79828   5.182 9.08e-05 ***
#as.factor(x)2    5.72079    2.17955   2.625  0.01839 *
#z                0.47615    0.12865   3.701  0.00194 **
#as.factor(x)2:z -0.09588    0.18195  -0.527  0.60544
#---
#Signif. codes:  0 ‘***’ 0.001 ‘**’ 0.01 ‘*’ 0.05 ‘.’ 0.1 ‘ ’ 1
#
#Residual standard error: 1.169 on 16 degrees of freedom
#Multiple R-squared:  0.9522,   Adjusted R-squared:  0.9432
#F-statistic: 106.3 on 3 and 16 DF,  p-value: 8.896e-11

# Predict for x = 1, and y = 1:5
predict(fit, newdata = data.frame(x = 1, z = 1:5));
#1        2        3        4        5
#4.613097 5.089242 5.565388 6.041533 6.517679

请注意，如果您想要 predict 基于预测变量新值的响应，您需要提供 newdata data.frame。否则，predict 将根据您的原始数据预测响应。

Answer 2

a <- lm(y ~ factor(x) + z + factor(x)*z, data=dataset)
df <- data.frame(x = c(1,1,1), z = c(1,2,3))
predict(a, df)

上面的想法是创建一个数据框，其中包含要测试模型的 X 和 Z 的值。

当回归有一个因素时 R 中的预测值

Predicted values in R when regression has a factor

regression

r

predict