Naive Bayes Classifier Python Program