正确使用R naive_bayes()和predict()

时间:2018-05-26 17:22:21

标签: r naivebayes

我正在尝试运行一个简单朴素的贝叶斯模型(尝试重做我所看到的数据广播课程)。

我正在使用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 = "prob"

我不确定我做错了什么。

0 个答案:

没有答案