Machine learning Models Used

  • Natural Language Processing (NLP) - Bidirectional Encoder Representations from Transformers (BERT) and BERTweet (A pre-trained language model for English Tweets)
  • Convolutional Neural Networks (CNN)
  • CNN with Transfer Learning:
    • VGG16, VGG19, ResNet50, ResNet152, ResNet152V2
    • DenseNet201, Xception, InceptionV3, EfficientNetB7, MobileNetV
  • Long-Short Term Memory (LSTM) Recurrent Neural Networks (RNN)
  • XGBoost
  • Decision Trees
  • Random Forest
  • Simple Linear/Multivariate Regression
  • Logistic Regression
  • General Linear Model (GLM)
  • K-Means Clustering
  • Seasonal ARIMA
  • STL Decomposition
  • Exponential Smoothing
  • Prophet