Predicting classes based on Naive Baye's Classifier