将决策边界拟合到 R 中的逻辑回归模型

我正在努力使用 ggplot 在 R 中绘制决策边界。

我有两个变量（考试成绩）和一个学生是否被录取的二元分类。数据如下所示：

> head(exam.data)
  Exam1Score Exam2Score Admitted
1   34.62366   78.02469        0
2   30.28671   43.89500        0
3   35.84741   72.90220        0
4   60.18260   86.30855        1
5   79.03274   75.34438        1
6   45.08328   56.31637        0
  

我可以使用 ggplot 绘制数据：

exam.plot <- ggplot(data=exam.data, aes(x=Exam1Score, y=Exam2Score, col = ifelse(Admitted == 1,'dark green','red'), size=0.5))+
  geom_point()+
  labs(x="Exam 1 Scores", y="Exam 2 Scores", title="Exam Scores", colour="Exam Scores")+
  theme_bw()+
  theme(legend.position="none")

然后成功拟合逻辑回归模型：

exam.lm <- glm(data=exam.data, formula=Admitted ~ Exam1Score + Exam2Score, family="binomial") 

因此，在大量搜索网络后，我决定手动拟合决策边界（尽管确实尝试使用 stat_smooth 执行此操作一段时间，但无法使其工作），我尝试了以下操作：

# Fit the decision boundary
plot_x <- c(min(exam.data$Exam1Score)-2, max(exam.data$Exam1Score)+2)
plot_y <- (-1 /coef(exam.lm)[3]) * (coef(exam.lm)[2] * plot_x + coef(exam.lm)[1])
db.data <- data.frame(rbind(plot_x, plot_y))
colnames(db.data) <- c('x','y')

# Add the decision boundary plot
ggplot()+geom_line(data=db.data, aes(x=x, y=y))

它成功地绘制了决策边界，但我无法将其添加到我现有的图中：

> exam.plot+geom_line(data=db.data, aes(x=x, y=y))
Error: Aesthetics must either be length one, or the same length as the dataProblems:x, y

有人可以指出我做错了什么或者我是否真的可以用 +stat_smooth() 做到这一点？

所有代码（ex2.R）和文件都在这里：https://github.com/StuHorsman/rscripts/tree/master/R/Coursera https://github.com/StuHorsman/rscripts/tree/master/R/Coursera

Thanks!

Stuart

更新：我可以通过以下方式实现一些类似的目标：

plot(exam.data$Exam1Score, exam.data$Exam2Score, type="n", xlab="Exam 1 Scores", ylab="Exam 2 Scores")      
points(exam.data$Exam1Score[exam.data$Admitted==1], exam.data$Exam2Score[exam.data$Admitted==1], pch=4, col="green")  
points(exam.data$Exam1Score[exam.data$Admitted==0], exam.data$Exam2Score[exam.data$Admitted==0], pch=4, col="red")        
lines(db.data, col="blue")

问题是在exam.plot你不仅仅使用美学x and y， 但是也col and size（后者不必要）。这些层需要有all中定义的美学集ggplot ()称呼。 （我经常被这个问题困扰）。

Thus:

exam.plot+geom_line(data=db.data, aes(x=x, y=y), col = "black", size = 1)

确实有情节。

不过，我建议改变exam.plot一点并删除不适用于所有图层的所有美学（并将它们放入图层定义中）：

exam.plot <- ggplot(data=exam.data, aes(x = Exam1Score, y=Exam2Score))+
  geom_point(aes (col = Admitted), size = 0.5)+
  scale_color_manual (values =  c('red', 'dark green')) + 
  labs(x="Exam 1 Scores", y="Exam 2 Scores", title="Exam Scores", colour="Exam Scores")+
  theme_bw()+
  coord_equal () +  # assuming that the scores have the same scale.
  theme(legend.position="none")

exam.plot + geom_line(data=db.data, aes(x=x, y=y))

其中有示例数据

exam.data <- data.frame (Exam1Score = rnorm (100) + 0:1, 
                         Exam2Score = rnorm (100) + 0:1, 
                         Admitted = factor (rep (0:1, 50)))

yields:

（使用默认大小绘制，在本例中几乎看不到 0.5）