train test split for nltk naive bayes