Как сделать, чтобы мой вывод SVM отображал прогнозируемый тип класса вместо фактического числового значения? - PullRequest
Новая
       21

Как сделать, чтобы мой вывод SVM отображал прогнозируемый тип класса вместо фактического числового значения?

0 голосов
/ 16 февраля 2020

Я делаю SVM для R, и когда я делаю: 

library(ISLR)
library(e1071)

fit2 = svm(satisfied~PC1 + PC2 + PC3 + PC4, data=data.frame(train.incl.score.x.y.st), kernel="radial", cost=5, gamma=500)

pred2=predict(fit2, data.frame(val.incl.score[,1:4]),  probability=FALSE)

table(pred2, data.frame(val.incl.score.x.y.st)$satisfied)

>> pred2.                  0 1
  -8.37175892132423        1 0
  -7.98589263979825        1 0
  -6.74777840420131        1 0
  -6.70935986153772        1 0
  -6.46316720580385        1 0
  -6.04268844390794        1 0

 <list goes on...>

Я хочу, чтобы столбец pred2 в выводе выше отображал прогнозируемый тип класса вместо числового значения, которое Модель SVM присваивает каждому наблюдению. Как я могу получить прогнозируемый тип класса для каждого наблюдения валидации и соответственно создать матрицу путаницы?

Спасибо,

...