未显示 ggplot 上的回归线

Question

当我尝试使用 ggplot 时，该图仅以点显示数据，但图中根本没有线。此外，R 中没有错误。数据有两个列，即月份和降雨量。我做了几年的数据集如下：

Month Rainfall
1        0.7
2         0
3         0
.         .
.         .
12         1.2
1         0
2         0.2
.         .
.         .

我的项目的ggplot完整代码如下：

 split = sample.split(dataset$Rainfall, SplitRatio = 0.8)
 training_set = subset(dataset, split == TRUE)
 test_set = subset(dataset, split == FALSE)


 regressor = lm(formula = Rainfall ~ Month,
                data = training_set)

 y_pred = predict(regressor, newdata = test_set)
 y_pred


 library(ggplot2)

 ggplot() + 
   geom_point(aes(x = training_set$Month, y = training_set$Rainfall),
               color = 'red') +
   geom_line(aes(x = training_set$Month, y = predict(regressor, newdata = training_set)),
               color = 'blue') +
   ggtitle('Rainfall (Training set)') +
   xlab('Month') +
   ylab('Rainfall')

 ggplot() + 
   geom_point(aes(x = test_set$Month, y = test_set$Rainfall),
               color = 'red') +
   geom_line(aes(x = training_set$Month, y = predict(regressor, newdata = training_set)),
               color = 'blue') +
   ggtitle('Monthly Rainfall (Test set)') +
   xlab('Month') +
   ylab('Rainfall')

但是，我不能将一条线绘制为简单的线性回归。

Answer 1

对于 ggplot2，您可以使用 geom_smooth(method="lm") 绘制简单的线性回归线。

你可以参考这个https://github.com/rstudio/cheatsheets/raw/master/data-visualization-2.1.pdf备忘单了解如何使用ggplot2。

---

您的代码的“固定”版本示例：

library(tidyverse)
test_set %>% ggplot(aes(x = Month, y = Rainfall)) + 
   geom_point() +
   geom_smooth(method="lm", se=FALSE) +
   ggtitle('Monthly Rainfall (Test set)') +
   xlab('Month') +
   ylab('Rainfall')

我用这个作为测试集

test_set <- tribble(
  ~Month, ~Rainfall,
  1,   1,
  2,   2,
  3,   3,
  4,   2,
  5,  .7
)