如何使用 R 一起绘制训练和测试？

Question

我想将训练集和测试集绘制在同一张图中，以便添加预测线，我所能做的就是

 prev <- as.data.frame(predict(m,newdata=as.data.frame(test)))
plot(train,ylab="Vol",type="l",col="red")
lines(prev,type="l",col="blue")
lines(test,type="l",col="yellow")

Answer 1

查看此资源：adding regression line per group with ggplot2

一般来说，您可以使用 ggplot2、geom_point 和 geom_line/geom_smooth 在一张图中绘制多个数据集和线条。在不知道您的数据的情况下，我将采取以下步骤

1. 添加一列指定数据是训练还是测试：

train$source <- "train"
test$source <- "test"

1. 合并数据集：

data <- (rbind(train, test))

1. 使用 ggplot、geom_point() 和 geom_smooth()/geom_line()

ggplot(data, aes(x=yourxvar, y=Vol, color=factor(source))) +
geom_point() + geom_smooth(method="lm")

您必须填写一些确切的代码和数据。我会假设 train 和 test 是您观察到的数据框，但是您似乎将它们视为线条？您只想绘制预测线以进行测试还是同时进行训练？无论哪种方式，我都会将其命名为比 prev 更具信息性的名称。