分类数据的朴素贝叶斯分类

时间:2013-08-06 10:37:39

标签: r machine-learning classification bayesian-networks categorical-data

我是统计和数据挖掘的新手。我遵循了here的例子,这完美无缺。现在我想将此方法应用于我的数据集,但该数据集仅包含分类数据。

R出现以下错误:

Error in train.default(x, y, "nb", trControl = trainControl(method = "cv",  :
wrong model type for regression

我的问题是:为了将该方法应用于我的分类数据集,我应该进行哪种转换?

CODE SNIPPET

bbb <- data.frame(x=c("Cleveland", "London", "London", "Ashburn"),y= c("USA", "GBR", "GBR", "USA"))
x = bbb$x
y = bbb$y
model = train(x,y,'nb',trControl=trainControl(method='cv',number=1))
model

1 个答案:

答案 0 :(得分:1)

因为数据集的标签是数字格式,R决定使用回归而不是分类。

将数据集的标签更改为字符而不是数字。所以R不会混淆。