您无法使用单个列联表生成完整的 ROC 曲线,因为列联表仅提供单个灵敏度/特异性对(对于用于生成列联表的任何预测截止值)。
如果您有许多使用不同截止值生成的列联表,您将能够近似 ROC 曲线(基本上它将是列联表中灵敏度/特异性值之间的线性插值)。例如,让我们考虑使用逻辑回归预测鸢尾花数据集中的花是否杂色:
iris$isv <- as.numeric(iris$Species == "versicolor")
mod <- glm(isv~Sepal.Length+Sepal.Width, data=iris, family="binomial")
我们可以使用标准的ROCR 代码来计算这个模型的 ROC 曲线:
library(ROCR)
pred1 <- prediction(predict(mod), iris$isv)
perf1 <- performance(pred1,"tpr","fpr")
plot(perf1)
现在让我们假设我们所拥有的不是mod,而是带有多个预测截止值的列联表:
tables <- lapply(seq(0, 1, .1), function(x) table(iris$isv, factor(predict(mod, type="response") >= x, levels=c(F, T))))
# Predict TRUE if predicted probability at least 0
tables[[1]]
# FALSE TRUE
# 0 0 100
# 1 0 50
# Predict TRUE if predicted probability at least 0.5
tables[[6]]
# FALSE TRUE
# 0 86 14
# 1 29 21
# Predict TRUE if predicted probability at least 1
tables[[11]]
# FALSE TRUE
# 0 100 0
# 1 50 0
从一个表到下一个表,由于截止值增加,一些预测从 TRUE 变为 FALSE,通过比较连续表的第 1 列,我们可以确定其中哪些代表真阴性和假阴性预测。遍历我们的列联表的有序列表,我们可以创建可以传递给 ROCR 的假预测值/结果对,确保我们匹配每个列联表的敏感性/特异性。
fake.info <- do.call(rbind, lapply(1:(length(tables)-1), function(idx) {
true.neg <- tables[[idx+1]][1,1] - tables[[idx]][1,1]
false.neg <- tables[[idx+1]][2,1] - tables[[idx]][2,1]
if (true.neg <= 0 & false.neg <= 0) {
return(NULL)
} else {
return(data.frame(fake.pred=idx,
outcome=rep(c(0, 1), times=c(true.neg, false.neg))))
}
}))
现在我们可以像往常一样将伪造的预测传递给 ROCR:
pred2 <- prediction(fake.info$fake.pred, fake.info$outcome)
perf2 <- performance(pred2,"tpr","fpr")
plot(perf2)
基本上我们所做的是对 ROC 曲线上的点进行线性插值。如果您有许多截止值的列联表,您可以更接近真实的 ROC 曲线。如果您没有广泛的截止范围,您就无法希望准确地再现完整的 ROC 曲线。