(BOW, TF-IDF, Word2Vec, BERT) Word Embeddings + (SVM, Naive Bayes, Decision Tree, Random Forest) Base Classifiers + Pre-trained BERT on Tensorflow Hub + 1-D CNN and Bi-Directional LSTM on IMDB Movie Reviews Dataset
Rank | Word Embedding | Classifier | Accuracy | F1-Score |
---|---|---|---|---|
1 | BERT Sentence Version (Mean Bert Features per Review) | SVM | 90.35 | 0.90 |
2 | BERT Sentence Version (Mean Bert Features per Review) | MLP | 90.32 | 0.90 |
3 | TFIDF with Stop Words | SVM | 89.59 | 0.90 |
Rank | Word Embedding | Model | Accuracy |
---|---|---|---|
1 | BERT TensorFlow-HUB | Bi-Directional LSTM | 91.34 |
2 | BERT Sentence Version (Mean Bert Features per Review) | 1-D CNN | 85.46 |