Sentiment DistilBERT (IMDB)
Classificatore binario di sentiment (NEGATIVE/POSITIVE) ottenuto dal fine-tuning di distilbert-base-uncased.
Dati
Sottoinsieme di IMDB: 2000 esempi di training, 500 di test (seed 42).
Addestramento
lr 2e-5, 3 epoche, batch 16, fp16, GPU T4.
Metriche
Accuracy sul test (500 esempi): 86.8%
Limiti
- Addestrato solo su recensioni di film in inglese: non affidabile su altre lingue o domini.
- Poche risorse di training (2000 esempi): non adatto a uso in produzione.
- Può riprodurre bias presenti nel dataset; ironia e sarcasmo sono difficili.
- Downloads last month
- 20
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support
Model tree for lucamaie/sentiment-distilbert-imdb
Base model
distilbert/distilbert-base-uncased