glm()模型的交叉验证

时间:2014-01-27 11:56:05

标签: r partitioning prediction glm cross-validation

我正在尝试对我之前在R中构建的一些glm模型进行10倍交叉验证。我对cv.glm()包中的boot函数有点困惑,虽然我已经阅读了很多帮助文件。当我提供以下公式时:

library(boot)
cv.glm(data, glmfit, K=10)

这里的“数据”参数是指整个数据集还是仅指测试集?

到目前为止我看到的例子提供了“数据”参数作为测试集,但这并没有真正意义,例如为什么在同一个测试集上有10倍?它们都会给出完全相同的结果(我假设!)。

不幸的是?cv.glm以模糊的方式解释它:

  

data:包含数据的矩阵或数据框。行应该是   case和列对应变量,其中一个是变量   响应

我的另一个问题是$delta[1]结果。这是10次试验的平均预测误差吗?如果我想获得每个折叠的错误怎么办?

这是我的脚本的样子:

##data partitioning
sub <- sample(nrow(data), floor(nrow(x) * 0.9))
training <- data[sub, ]
testing <- data[-sub, ]

##model building
model <- glm(formula = groupcol ~ var1 + var2 + var3,
        family = "binomial", data = training)

##cross-validation
cv.glm(testing, model, K=10)

2 个答案:

答案 0 :(得分:14)

我总是对使用各种包10倍交叉验证方法有点谨慎。我有自己的简单脚本来手动为任何机器学习包创建测试和训练分区:

#Randomly shuffle the data
yourData<-yourData[sample(nrow(yourData)),]

#Create 10 equally size folds
folds <- cut(seq(1,nrow(yourData)),breaks=10,labels=FALSE)

#Perform 10 fold cross validation
for(i in 1:10){
    #Segement your data by fold using the which() function 
    testIndexes <- which(folds==i,arr.ind=TRUE)
    testData <- yourData[testIndexes, ]
    trainData <- yourData[-testIndexes, ]
    #Use test and train data partitions however you desire...
}

答案 1 :(得分:6)

@Roman在他的评论中提供了一些答案,但是,通过cv.glm检查代码来提供问题的答案:

我相信这段代码会将数据随机分成K-folds,如果K不分割n,则根据需要排列舍入:

if ((K > n) || (K <= 1)) 
    stop("'K' outside allowable range")
K.o <- K
K <- round(K)
kvals <- unique(round(n/(1L:floor(n/2))))
temp <- abs(kvals - K)
if (!any(temp == 0)) 
    K <- kvals[temp == min(temp)][1L]
if (K != K.o) 
    warning(gettextf("'K' has been set to %f", K), domain = NA)
f <- ceiling(n/K)
s <- sample0(rep(1L:K, f), n)

这里的这一位表明delta值不是均方根误差。正如帮助文件所说The default is the average squared error function.这是什么意思?我们可以通过检查函数声明来看到这一点:

function (data, glmfit, cost = function(y, yhat) mean((y - yhat)^2), 
    K = n) 

表明在每个折叠中,我们计算误差平方的平均值,其中误差是通常意义上的预测响应与实际响应之间的差异。

delta[1]只是每个折叠的所有这些字词的SUM的weighted average,请参阅cv.glm代码中的内联注释:

for (i in seq_len(ms)) {
    j.out <- seq_len(n)[(s == i)]
    j.in <- seq_len(n)[(s != i)]
    Call$data <- data[j.in, , drop = FALSE]
    d.glm <- eval.parent(Call)
    p.alpha <- n.s[i]/n #create weighted average for later
    cost.i <- cost(glm.y[j.out], predict(d.glm, data[j.out, 
        , drop = FALSE], type = "response"))
    CV <- CV + p.alpha * cost.i # add weighted average error to running total
    cost.0 <- cost.0 - p.alpha * cost(glm.y, predict(d.glm, 
        data, type = "response"))
}