向线图R添加多条回归线

Question

这是我正在使用的线性模型：

blk.lm <- lm(formula = YEARS ~ AGE + AGE2, data = BLKFRIDAY)

我想用ggplot2创建一个图，该图将YEARS的{​​{1}}与AGE的{​​{1}}进行比较，其中YEARS只是{ {1}}。我的目标是查看线性模型还是二次模型最适合我的数据。

如何设置一个图，其中有两条回归线，分别是AGE2，然后是AGE2以进行比较？

Answer 1

以下是diamonds中的ggplot2数据集的示例：

library(ggplot2)
ggplot(diamonds, aes(x = carat, y =  price)) + 
  geom_point() +
  geom_smooth(formula = y ~ x, se = FALSE, method = "lm") +
  geom_smooth(formula = y ~ I(x^2), se = FALSE, method = "lm", color = "red")
# I(x^2) is needed because x^2 is interpreted as x * X - the interaction of x with x

请注意，红线是弯曲的，因为它相对于carat是非线性的。