---
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](https://www.kaggle.com/datasets/lakshmi25npathi/imdb-dataset-of-50k-movie-reviews)
- 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