Naive Bayes Classification for Categorical Data

Question

I am new to statistics and data mining. I followed the example here, which worked perfect. Now I want to apply this method to my dataset which, however, consists of categorical data only.

R gives the following error:

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

My question is: what sort of transformations should I make in order to apply that method on my categorical set of data?

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

Please provide a [reproducible example](http://stackoverflow.com/questions/5963269/how-to-make-a-great-r-reproducible-example). — Thomas, Aug 06 '13 at 11:40
ok guys, I have updated the question so as to show you what I have done so far. — user2295350, Aug 06 '13 at 12:38
This appears to work: `model = train(bbb$x,bbb$y,'nb',trControl=trainControl(number=1))`. Maybe `method='cv'` is not a good fit for this toy problem? — Doctor Dan, Aug 06 '13 at 14:55

Shahmohamamdi · Answer 1 · 2015-02-06T13:46:38.293

1

because labels of your dataset are in numeric format, R decide to use regression instead of classification.

change labels of data set to characters instead of numbers. so R will not confuse.

edited Feb 06 '15 at 13:46

answered Feb 06 '15 at 13:19

Shahmohamamdi

111
2

Naive Bayes Classification for Categorical Data

1 Answers1