Naive Bayes Model Code