我 Build 了一个简单的线性回归模型,并使用该模型产生了一些预测值 . 但是,我更感兴趣的是在图表上可视化它,但我不知道如何添加图例以突出显示原始mpg值'black'和新预测值为"red" .
此示例中使用的数据是来自数据集包的mtcars数据集
library(ggplot2)
library(datasets)
library(broom)
# Build a simple linear model between hp and mpg
m1<-lm(hp~mpg,data=mtcars)
# Predict new `mpg` given values below
new_mpg = data.frame(mpg=c(23,21,30,28))
new_hp<- augment(m1,newdata=new_mpg)
# plot new predicted values in the graph along with original mpg values
ggplot(data=mtcars,aes(x=mpg,y=hp)) + geom_point(color="black") + geom_smooth(method="lm",col=4,se=F) +
geom_point(data=new_hp,aes(y=.fitted),color="red")
散点图
2 回答
这是一个想法 . 您可以将预测数据和观察数据组合在同一数据框中,然后创建散点图以生成图例 . 以下代码是现有代码的扩展 .
这是另一种没有
dplyr
的方法: