stanfordnlp/imdb
Viewer • Updated • 100k • 186k • 467
Binary sentiment classifier (positive/negative) fine-tuned from distilbert-base-uncased.
from transformers import pipeline
classifier = pipeline("sentiment-analysis", model="marksoulier/hf-workshop-sentiment")
classifier("I loved this film!")
# [{'label': 'POSITIVE', 'score': 0.998}]
Training code, dependencies, and full reproduction steps are on GitHub: https://github.com/marksoulier/hf-workshop
git clone https://github.com/marksoulier/hf-workshop
cd hf-workshop
uv run train.py
Base model
distilbert/distilbert-base-uncased