如何访问 R 中 predict() function 的值进行存储？

Question

我正在为 class 训练一个 K-最近邻 model。问题是他们要求我们用整个数据库训练它，除了被预测的行。

我的计划是初始化一个用于存储的向量，并运行一个 for 循环以遍历每一行，省略特定行进行训练，然后将预测值附加到向量，并在循环后计算精度：

results <- c()
for (i in nrow(data) {
        model.kknn <- train.kknn(data[-i,11]~., data = data[-i,1:10],kmax = 7, scale = TRUE)
        pred <- predict(model,data[i,1:10])
        results <- c(results,pred)
}

我期望矢量results是一系列 1 和 0。 但是，我尝试只循环第一行， results的值为2 。

打印pred时的值为：

[1] 1

Levels: 0 1

知道如何将 1 到 append 转换为矢量results吗？

Answer 1

在 for() 部分指定1:N ，最好不要“增长”一个向量，而是初始化一个适当长度的空向量并填充它。

N <- nrow(data)
results <- vector(length=N)
for (i in 1:N) {
        model.knn <- train.kknn(data[-i,11]~., data=data[-i,1:10], kmax=7, scale=T)
        results[i] <- predict(model.knn, data[i,1:10,drop=F])
}