README.md

metadata

base_model:
  - google-bert/bert-base-uncased
tags:
  - sentiment-analysis
  - text-classification
  - imdb
  - bert
  - fine-tuned
language:
  - en
pipeline_tag: text-classification

BERT IMDb Sentiment Classifier

Dataset

• IMDb Dataset (50,000 reviews: 25k train + 25k test)
• Source: IMDB Dataset of 50K Movie Reviews Kaggle
• Preprocessing: Tokenization via BERT tokenizer

Model Architecture

• Base: bert-base-uncased from Hugging Face Transformers
• Fine-tuned with a classification head (linear layer)
• Trained with PyTorch using AdamW optimizer and scheduler