Правильное использование R naive_bayes () и Forect () - PullRequest
0 голосов
/ 26 мая 2018

Я пытаюсь запустить простую наивную байесовскую модель (пытаюсь повторить то, что я видел в курсе datacamp).

Я использую пакет R naivebayes.

Обучениенабор данных where9am выглядит следующим образом:

where9am

Моя первая проблема заключается в следующем ... когда у меня есть несколько прогнозов в кадре данных thursday9am ...

thursday9am

... и я использую следующий код:

locmodel <- naive_bayes(location ~ daytype, data = where9am)
my_pred <- predict(locmodel, thursday9am)

Я получаю серию <NA>, в то время как она хорошо работает с правильным прогнозом, еслиthursday9am Фрейм данных содержит только одно наблюдение.

output with multiple rows to predict

output with a single prediction

Вторая проблема заключается в следующем: когда я использую следующий код для получениясвязанные вероятности ...

locmodel <- naive_bayes(location ~ daytype, data = where9am, type = c("class", "prob"))
predict(locmodel, thursday9am , type = "prob")

... даже если у меня есть только одно наблюдение в thursday9am, я получаю серию <NaN>.

output with type =

Я не уверен, что я делаю неправильно.

...