This is part of the Data Science Lab course. We have done classification of emails as spam or ham using various Naive Bayes models. And comparing it with RNN based model LSTM. We fine-tuned a version of BERT model for the same task and compared results. The list of models are as follows;
- Gaussian Naive Bayes
- Multinomial Naive Bayes
- Binomial Naive Bayes
- Uni-directional
- Bi-directional
The research paper we used for our work : https://www.sciencedirect.com/science/article/pii/S1877050921007493