Какой метод лучше всего предсказать результат с 6 вариантами? - PullRequest
0 голосов
/ 26 апреля 2019

Я хотел бы предсказать «Ранжирование» (1-6) одной из моих 6 кофемашин с личностными качествами «Большой пятерки» (ОКЕАН), и я действительно не знаю, какие алгоритмы были бы лучшими для этой задачи. Любая рекомендация будет отличной помощью для меня. Я пытался предсказать это с помощью C5.0, но результат был довольно плохим, поэтому я подумал, что, может быть, есть лучший способ или метод, чтобы предсказать это. Или, может быть, я делаю что-то не так, потому что я довольно плохо знаком с машинным обучением и прогнозированием.

Моя таблица выглядит так: (116 записей) enter image description here

Все столбцы "BFI_" являются числовыми и имеют диапазон от 1 до 5.
Колонка «Philips» (Рейтинг для кофемашины Philips) имеет тип данных (1-6). .

ind <- sample(2, nrow(dfrandomzied), replace = T, prob = c(0.7, 0.3)) 
set.seed(100)    
train <- df[ind == 1,]

test <- df[ind == 2,]

# Building the model
m1 <- C5.0(train[,1:5], train$Philips) 
summary(m1)

# Prediction with Testdaten
p1 <- predict(m1, test)

Сводка m1

enter image description here

Прогноз p1 результаты

# Prediction with Testdaten
p1 <- predict(m1, test)

# Auswertung Confusionmatrix
confusionMatrix(p1, test$Philips)

enter image description here

Любой совет приветствуется :). Заранее спасибо.!

...