r - 使用随机森林创建二元结果

Question

我有一个如下所示的数据集:

 TEAM1         TEAM2     EXPG1  EXPG2         Gewonnen    
 ADO Den Haag  Groningen 1.5950 1.2672        1

我现在尝试根据 EXPG1 和 EXPG2 预测列 Gewonnen。因此，我创建了一个训练和测试集，并创建了以下模型(全部使用 rcaret):

modFit <- train(Gewonnen~ EXPG1 + EXPG2, data=training, method="rf", prox=TRUE)

我现在无法制作混淆矩阵，因为我的数据有更多引用。这是真的，因为当我这样做时:

pred <- predict(modFit, testing)
head(print)

它说:0.5324000 0.7237333 0.2811333 0.8231000 0.8299333 0.9792000

因为我想制作一个混淆矩阵，所以无法将它们转换为 0/1，但我感觉模型中也应该有一个选项可以执行此操作。

关于我应该在此模型中更改哪些内容以创建 0/1 值的任何想法。我在文档中找不到它:

modFit <- train(Gewonnen~ EXPG1 + EXPG2, data=training, method="rf", prox=TRUE)

Answer 1

首先，正如 Tim Biegeleisen 所说，您应该将您的 Gewonnen 变量转换为一个因子(在训练和测试集中)，如果它还没有:

training$Gewonnen <- as.factor(training$Gewonnen)
testing$Gewonnen <- as.factor(testing$Gewonnen)

之后，caret 函数 predict 中的 type 选项决定了二元分类问题得到的响应类型，即 class标签或概率。这是 caret documentation 中的可重现示例使用 mlbench 包中的 Sonar 数据集:

library(caret)
library(mlbench)
data(Sonar)
str(Sonar$Class)
# Factor w/ 2 levels "M","R": 2 2 2 2 2 2 2 2 2 2 ...

set.seed(998)
inTraining <- createDataPartition(Sonar$Class, p = .75, list = FALSE)
training <- Sonar[ inTraining,]
testing  <- Sonar[-inTraining,]

modFit <- train(Class ~ ., data=training, method="rf", prox=TRUE)

pred <- predict(modFit, testing, type="prob") # for class probabilities
head(pred)
#        M     R
# 5  0.442 0.558
# 10 0.276 0.724
# 11 0.096 0.904
# 12 0.360 0.640
# 20 0.654 0.346
# 21 0.522 0.478

pred2 <- predict(modFit, testing, type="raw") # for class labels
head(pred2)
# [1] R R R R M M
# Levels: M R

对于混淆矩阵，您将需要类标签(即上面的 pred2):

confusionMatrix(pred2, testing$Class)
# Confusion Matrix and Statistics

#           Reference
# Prediction  M  R
#          M 25  6
#          R  2 18