Как получить, какой результат удовлетворяет прогнозу путаницы - PullRequest
0 голосов
/ 04 декабря 2018

Я сделал модель дерева решений, чтобы предсказать MVP из набора данных игрока NBA.Результат матрицы путаницы показывает, что у нее есть три правильных предсказания, поэтому мне интересно узнать, какие три игрока из моего набора данных в любом случае?

enter image description here

1 Ответ

0 голосов
/ 04 декабря 2018

Поскольку вы создали матрицу путаницы, я уверен, что у вас уже есть тестовые метки.Просто найдите индексы тестовых меток, которые соответствуют классу 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],])
}
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...