Skip to content

(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

Notifications You must be signed in to change notification settings

hmohebbi/SentimentAnalysis

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 

Repository files navigation

SentimentAnalysis

(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

Results for Base Classifiers

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

Results for Deep Neural Networks

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

About

(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

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published