Ошибка R: все аргументы должны иметь одинаковую длину

Question

Я получил ошибку, когда я делаю наивный байесовский от R, вот мой код и ошибка

library(e1071) 

#data

train_data <- read.csv('https://raw.githubusercontent.com/JonnyyJ/data/master/train.csv',header=T)
test_data <- read.csv('https://raw.githubusercontent.com/JonnyyJ/data/master/test.csv',header=T)      

efit <- naiveBayes(y~job+marital+education+default+contact+month+day_of_week+
                        poutcome+age+pdays+previous+cons.price.idx+cons.conf.idx+euribor3m
                       ,train_data)  

pre <- predict(efit, test_data)
bayes_table <- table(pre, test_data[,ncol(test_data)])
accuracy_test_bayes <- sum(diag(bayes_table))/sum(bayes_table)
    list('predict matrix'=bayes_table, 'accuracy'=accuracy_test_bayes)

ОШИБКА:

bayes_table<- таблица (pre, test_data [, ncol (test_data)]) Ошибка в таблице (pre, test_data [, ncol (test_data)])): все аргументы должны иметь одинаковую длину precision_test_bayes <- sum (diag (bayes_table)) / sum (bayes_table) Ошибка в diag (bayes_table): объект 'bayes_table' не найден в списке ('матрица прогнозирования' = bayes_table, 'precision' = precision_test_bayes) Ошибка: объект 'bayes_table' не найден

Я действительно не понимаю, что происходит, потому что я новичок в R

Answer 1

По какой-то причине по умолчанию predict(efit, test_data, type = "class") не работает в этом случае (возможно, потому что ваша модель предсказывает 0 для всех наблюдений в тестовом наборе данных). Вам также нужно построить таблицу, используя ваш результат (т.е. test_data[,ncol(test_data)] возвращается euribor3m). Следующее должно работать:

pre <- predict(efit, test_data, type = "raw") %>%
  as.data.frame() %>%
  mutate(prediction = if_else(0 < 1, 0, 1)) %>%
  pull(prediction)

bayes_table <- table(pre, test_data$y)

accuracy_test_bayes <- sum(diag(bayes_table)) / sum(bayes_table)

list('predict matrix' = bayes_table, 'accuracy' = accuracy_test_bayes)
# $`predict matrix`
#    
# pre    0    1
#   0 7282  956
# 
# $accuracy
# [1] 0.8839524