Amazon Reviews Dataset obtained from Kaggle
Machine learning models applied:
- Extreme Gradient Boosting
- Random Forest
- Gradient Boosting
- Naive Bayes
- Light GBM
- Decision Tree
After training machine learning models, a deep learning model, Bidirectional LSTM model is trained on the given dataset.
BI-LSTM performed best according to roc auc score of 0.89 as compared to other machine learning models.