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

Я использовал дерево решений, чтобы предсказать мой набор тестов.После запуска моего кода я получаю таблицу с результатами, но я хочу использовать команду confusionMatrix () из библиотеки caret.Я пробовал несколько вещей, но ни одна не сработала.Пожалуйста, смотрите мой код:

library(rpart)
tree <- rpart(train$number ~ ., train, method = "class")
pred <- predict(tree,test, type ="class")
p <- predict(tree, type="class")
# Confusion Matrix
conf <- table(test$number, pred)


> conf
                pred
                 Problem          Reference
  Problem              0            100
  Reference            0           2782

Я пытался сделать это:

p <- predict(tree, type="class")
confusionMatrix(p, entiredata$number)

Ошибки как данные и ссылки должны быть одного типа , поэтому я изменил оба коэффициента с as.factors(), тогда аргументы a были разной длины .Я искал в Интернете и нашел похожие вопросы, но все они не помогли мне.Моя конечная цель - получить статистику с точностью.

1 Ответ

0 голосов
/ 04 октября 2018
library(caret) 
confusionMatrix(p, test$number)

Поскольку вы predict только для данных испытаний, вам следует сравнивать прогнозы только для данных test, а не для всего набора данных.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...