【发布时间】:2018-01-09 17:05:22
【问题描述】:
我确实使用包klaR 将朴素贝叶斯应用于玩具数据集,一切正常。
接下来,我想使用method="nb" 复制与caret 相同的分析,这实际上只是klaR 包中NaiveBayes 函数的包装。
但是,后一种估计不起作用。我确实收到一条错误消息。
Error in NaiveBayes.default(x, y, usekernel = FALSE, fL = param$fL, ...) :
Zero variances for at least one class in variables: x1_disc_46, x1_disc_810, x2_disc_46
Timing stopped at: 0.01 0 0.02
我知道玩具数据集的局限性。然而,随着klaR 包的分析确实运行,我想知道如何使用caret 复制完全相同的分析?
代码如下:
# Data
d <- structure(list(Y = structure(c(1L, 1L, 1L, 1L, 2L, 2L, 2L, 2L,
2L, 1L), .Label = c("0", "1"), class = "factor"), x1_disc = structure(c(1L,
2L, 1L, 2L, 3L, 4L, 4L, 5L, 2L, 4L), .Label = c("_02", "_24",
"_46", "_68", "_810"), class = "factor"), x2_disc = structure(c(1L,
1L, 1L, 1L, 2L, 3L, 3L, 3L, 1L, 2L), .Label = c("_02", "_24",
"_46"), class = "factor")), .Names = c("Y", "x1_disc", "x2_disc"
), row.names = c(NA, -10L), class = "data.frame")
# Works(klaR)
library(klaR)
fit2 <- NaiveBayes(Y ~ x1_disc + x2_disc, usekernel = FALSE, fL = 0, data=d)
predict(fit2, d, threshold = 0)
# Does not work (caret)
library(caret)
model2 <- train(form=Y ~ x1_disc + x2_disc,
data=d,
method="nb",
# Uses package klaR
# (see: http://topepo.github.io/caret/train-models-by-tag.html)
trControl=trainControl(method="none"),
tuneGrid = data.frame(fL=0, usekernel=F, adjust=1))
predict(model2, d, type="prob")
【问题讨论】:
-
也许插入符号版本不接受“因素”类的数据。从“usekernal”而不是“usekernel”中的拼写错误,我假设您从交叉验证的问题中获得了代码。您是否按照他们的建议尝试不使用公式界面,而是指定
x=和y=?跨度> -
感谢您指出错字。我已经纠正了。接下来,我确实尝试指定
x和y,它成功了!这有点奇怪,但感谢您的快速帮助!
标签: r machine-learning r-caret naivebayes