姜瘟病
R语⾔回归中的Hosmer-Lemeshow以及calibrationcurve校正曲线Hosmer-Lemeshow test山药炖鸡汤
详细的Hosmer-Lemeshow test使⽤⽅法的介绍请参考:
calibration curve是评价模型的⼀个重要指标,不信的话请看,,总之很重要的
最近需要画校正曲线,但⽹上的相关资料实在是太少了,花了点时间研究了⼀下,跟⼤家分享⼀下如何使⽤Hosmer-Lemeshow test来绘制calibration curve,(p.s. 都是个⼈理解,可能有错,欢迎讨论):
⾸先,按照⽅法介绍中的样例,我们⽣成⼀下数据:
t.ed(43657)
n <- 100
x <- rnorm(n)
xb <- x
pr <- exp(xb)/(1+exp(xb))
y <- 1*(runif(n) < pr)
这⾥的x是变量,y相当于标签(0,1),这⾥做的是⼆分类的。然后我们放到glm函数中fit⼀下:
mod <- glm(y~x, family=binomial)
然后,将结果y和模型拟合概率传递给st函数,选择g = 10组。在这之前,要记得先安装并加载package,安装⽅法为
install.paskages(‘ResourceSelection’),加载⽅法为library(ResourceSelection)。
公务员平时考核hl <- st(mod$y, fitted(mod), g=10)
hl
这⾥的mod$y和 fitted(mod)分别表⽰标签和预测概率,可以换成⾃⼰的数据。hl输出为:
Hosmer and Lemeshow goodness of fit (GOF) test
data: mod$y, fitted(mod)
量尺寸
X-squared = 7.4866, df = 8, p-value = 0.4851
我们还可以从hl对象中获得⼀个观察到的与预期的表:
cbind(hl$obrved,hl$expected)
输出为
y0 y1 yhat0 yhat1
[0.0868,0.219] 8 2 8.259898 1.740102
(0.219,0.287] 7 3 7.485661 2.514339
(0.287,0.329] 7 3 6.968185 3.031815
(0.329,0.421] 8 2 6.194245 3.805755
(0.421,0.469] 5 5 5.510363 4.489637
(0.469,0.528] 4 6 4.983951 5.016049
(0.528,0.589] 5 5 4.521086 5.478914
(0.589,0.644] 2 8 3.833244 6.166756
(0.644,0.713] 6 4 3.285271 6.714729
(0.713,0.913] 1 9 1.958095 8.041905
其中y0, y1表⽰观察的结果,yhat0,yhat2表⽰预期的结果。为了理解这些数字都代表些什么,我们⼿动计算⼀下,⾸先,我们计算模型的预测概率,然后根据预测概率的⼗分位数对观察结果进⾏分组(这⾥其实就是将模型预测的概率进⾏排序,然后将数据划分成10组):
pihat <- mod$fitted
pihatcat <- cut(pihat, breaks=c(0,quantile(pihat, probs=q(0.1,0.9,0.1)),1), labels=FALSE)
接下来,我们循环遍历第1组到第10组,计算观察到的0和1的数量,并计算期望的0和1的数量。
这⾥提⼀下,上⾯我们选择了g=10,那么根据模型拟合出来的概率mod$fitted和⼗分位数,我们将所有数据分为10组,下⾯是本例中10组的分组情况,可以看到每个数据在哪个组别:
10 10 9 8 2 9 2 1 6 5 2 1 1 10 5 6 4 4 8 3 6 3 9 2 9 3 6 5 4 7 4 9 3 5 8 4 7 4 3 2 2 8 6 1 3 5 7 2 7 2 8 7 6 1 1 4 8 9 10 10 1 5 1 4 6 9 10 6 7 3 8 2 2 9 5 10 7 6 5 4 7 6 9 1 10 3 3 8 10 7 5 7 10 1 8 8 4 9 3 5
为了计算期望的0和1的数量,我们找到每个组中预测概率的平均值,然后将其乘以组⼤⼩,这⾥为10:
meanprobs <- array(0, dim=c(10,2))
expevents <- array(0, dim=c(10,2))
obvents <- array(0, dim=c(10,2))
stdprobs <- array(0, dim=c(10,2))
obsprobs <- array(0, dim=c(10,2))
for (i in 1:10) {
meanprobs[i,1] <- mean(pihat[pihatcat==i])
stdprobs[i,1] <- sd(pihat[pihatcat==i])
expevents[i,1] <- sum(pihatcat==i)*meanprobs[i,1]
obvents[i,1] <- sum(y[pihatcat==i])
宝宝咳嗽老不好怎么办呀
obsprobs[i,1] <- sum(y[pihatcat==i])/sum(pihatcat==i)
meanprobs[i,2] <- mean(1-pihat[pihatcat==i])
闺蜜的歌
stdprobs[i,2] <- sd(1-pihat[pihatcat==i])
expevents[i,2] <- sum(pihatcat==i)*meanprobs[i,2]
obvents[i,2] <- sum(1-y[pihatcat==i])
obsprobs[i,2] <- 1-(sum(y[pihatcat==i])/sum(pihatcat==i))
}
这⾥meanprobs 表⽰每组的平均概率,expevents 表⽰期望事件,obvents 表⽰观察事件,我加了两个参数,⽅便绘制calibration curve,其中,stdprobs 表⽰每组的标准差,obsprobs 表⽰每组的观察概率。
然后,我们可以通过上⾯计算得到的表格的10x2单元格中的(观察到的预期)^ 2 /预期的总和来计算Hosmer-Lemeshow检验统计量:
hosmerlemeshow <- sum((obvents-expevents)^2 / expevents)
hosmerlemeshow
可以看到输出为:
[1] 7.486643
和st函数计算出来的⼀样对吧。
calibration curve
唾液检测现在,我们有了观察的概率表obsprobs, 和预测的平均概率meanprobs ,以及预测概率的⽅差stdprobs,可以绘制calibration curves 了。
先绘制⼀个空图,
yy <- meanprobs[,c(1)]
xx <- obsprobs[,c(1)]
plot(xx, yy, type = "n", xlab = "Predicted probability", ylab = "Oberrved probability")
然后,添加数据,
points(xx, yy, type = "p", pch = 19, col = "agreen", lty = 1)
双线桥法通过改变type的值可以选择绘制散点图还是折线图,或者是别的类型的图。然后,我们在图上添加误差棒
plot_error <- function(xx, yy, sd, len = 1, col = "black") {
len <- len * 0.05
arrows(x0 = xx, y0 = yy, x1 = xx, y1 = yy - sd, col = col, angle = 90, length = len)
arrows(x0 = xx, y0 = yy, x1 = xx, y1 = yy + sd, col = col, angle = 90, length = len)
}
plot_error(xx, yy, sd = stdprobs[,c(1)], col = "agreen")
最后,加上对⾓线和legend,就可以绘制出⼀张校正曲线图啦。
abline(0,1,lty=3,lwd=1,col=c(rgb(0,0,0,maxColorValue=255)))
labs <- c("RANDOM TEST")
legend("top", legend = labs, cex = 0.8, lty = 1, lwd = 2, pch = 19, col = c("agreen"), int = 0.01, horiz = TRUE, l = "white")
P.S 这⾥使⽤的是简单的glm模型来做模型得到的概率,如果有现成的概率和标签,也是可以直接绘制calibration curve的~