标签: r decision-tree
我做了一个决策树模型,以从NBA球员数据集中预测MVP。混淆矩阵的结果表明它具有三个正确的预测,所以我仍然想从数据集中找出哪三个参与者吗?
答案 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],]) }