IMDB Sentiment Analyzer

A DistilBERT model fine-tuned to classify English movie reviews as positive or negative.

Results

  • Base model: distilbert-base-uncased
  • Dataset: stanfordnlp/imdb
  • Validation accuracy: 79.5%

Labels

  • LABEL_0: Negative
  • LABEL_1: Positive
Downloads last month
14
Safetensors
Model size
67M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Space using VENKATANISHNATH/imdb-sentiment-model 1