https://github.com/swap357/python-naive-bayes-classifier