train and test with naive bayes