Блог

У меня есть следующие 4 столбца: категория job1-категория работы первой работы человека, категория job2-категория работы второй работы человека и т. Д. Глядя только на людей, у которых была первая работа категории 1, я хочу иметь возможность предсказывать следующие категории должностей, используя наивные байесовские методы. Использование функции naive_bayes приводит к следующей ошибке.

 nb <- naive_bayes(data_train$job2category ~ ., data_train, usepoisson = TRUE)
 

 Error in naive_bayes.formula(data_train$job2category ~ ., data_train, : naive_bayes(): y has to be either a factor or character or logical vector
 

 structure(list(job1category = c(1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 
1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 
1, 1, 1, 1, 1, 1, 1, 1, 1), job2category = c(5, 5, 1, 2, 5, 1, 
1, 5, 2, 1, 5, 5, 1, 1, 1, 1, 5, 5, 1, 5, 1, 1, 1, 1, 1, 1, 1, 
5, 5, 1, 1, 1, 5, 1, 1, 1, 5, 5, 5, 5), job3category = c(5, 5, 
1, 2, 5, 1, 1, 5, 2, 1, 5, 5, 1, 1, 1, 1, 5, 5, 1, 5, 1, 1, 1, 
1, 1, 1, 1, 5, 5, 1, 1, 1, 5, 1, 1, 1, 5, 5, 5, 5), job4category = c(5, 
5, 1, 2, 5, 1, 1, 5, 2, 1, 5, 5, 1, 1, 1, 1, 5, 5, 1, 5, 1, 1, 
1, 1, 1, 1, 1, 5, 5, 1, 1, 1, 5, 1, 1, 1, 5, 5, 5, 5)), row.names = c(NA, 
-40L), class = c("tbl_df", "tbl", "data.frame"))
 

Я ценю любую помощь и открыт для получения советов о другом подходе к этому проекту.