DistilBERT Sentiment Classifier

DistilBERT fine-tuned for sentiment analysis on Rotten Tomatoes movie reviews.

Repository

sofiane-isi/distilbert-sentiment

Dataset

cornell-movie-review-data/rotten_tomatoes

Classes:

  • NEGATIVE
  • POSITIVE

Evaluation

  • Accuracy: 0.8461538461538461
  • Evaluation loss: 0.42963317036628723

Base model

distilbert-base-uncased

MLOps workflow

Dataset β†’ tokenization β†’ fine-tuning β†’ evaluation β†’ Hugging Face Hub β†’ Model Card β†’ Space.

Limitations

Educational binary sentiment classifier fine-tuned specifically on Rotten Tomatoes.

Downloads last month
23
Safetensors
Model size
67M params
Tensor type
F32
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Dataset used to train sofiane-isi/distilbert-sentiment

Space using sofiane-isi/distilbert-sentiment 1