File size: 663 Bytes
64cff59 40fdd30 b69efd1 40fdd30 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | ---
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
|