simple naive bayes made in python