将多条回归线添加到线图 R

Question

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

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

我想创建一个GGPLOT2剧情有比较YEARS来讲AGE到YEARS来讲AGE2 ，其中AGE2只是AGE^2 。 我的目标是查看线性模型还是二次模型最适合我的数据。

如何在Y ~ X1和Y ~ X2方面设置有两条回归线的图进行比较？

Answer 1

这是来自ggplot2的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是非线性的。