如何从混淆矩阵预测中获得满足的结果

时间:2018-12-04 02:32:50

标签: r decision-tree

我做了一个决策树模型,以从NBA球员数据集中预测MVP。混淆矩阵的结果表明它具有三个正确的预测,所以我仍然想从数据集中找出哪三个参与者吗?

enter image description here

1 个答案:

答案 0 :(得分:0)

既然您构建了混淆矩阵,我确信您已经有了测试标签。只需找到与MVP类相对应的测试标签的索引即可。在测试数据中使用这些索引来查找播放器的名称。您的代码应如下所示:

index<-which(test_label==1, arr.ind = TRUE) # assuming 1 is the class label for MVP
for iter in 1:dim(index)[1]{
print(test_data[index[iter],])
}