generated train and test data for naive bayes