Back to Projects

Sentiment Analysis on IMDB Reviews

Completed
ML/AI
Jun 2025

Overview

Multi-model sentiment classification on 50k IMDB reviews using classical ML, ANN, and BERT.

Highlights

  • EDA: class balance, review length distribution, word patterns; histograms + word clouds
  • Preprocessing: TF-IDF for NB/LR, Keras tokenization for ANN, HF tokenizer for BERT
  • Evaluated NB, Logistic Regression, ANN, BERT with F1/precision/recall/accuracy
  • Confusion matrices, PR curves for all models

Tech Stack

PythonPythonscikit-learnscikit-learnKerasKeras
HuggingFace Transformers

Related Projects