朴素貝葉斯分類數據分類

Question

我是統計和數據挖掘的新手。 我按照此處的示例進行了工作 ，效果很好。 現在，我想將此方法應用於我的數據集，該數據集僅由分類數據組成。

R給出以下錯誤：

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

我的問題是：為了將這種方法應用於我的分類數據集，我應該進行哪種轉換？

代碼片段

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

Answer 1

因為數據集的標簽為數字格式，所以R決定使用回歸而不是分類。

將數據集的標簽更改為字符而不是數字。 因此R不會混淆。