如何從混淆矩陣預測中獲得滿足的結果

Question

我做了一個決策樹模型，以從NBA球員數據集中預測MVP。 混淆矩陣的結果表明它具有三個正確的預測，因此我想知道我是否仍要從數據集中找出哪三個參與者？

Answer 1

由於您構建了混淆矩陣，因此我確定您已經有了測試標簽。 只需找到與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],])
}