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-learnKerasKerasHugging FaceHuggingFace Transformers

Related Projects