saiffff's picture
update
5db0e8f verified
metadata
library_name: transformers
tags:
  - sentiment-analysis
  - distilbert
  - imdb

Model Card for DistilBERT IMDb Sentiment Analysis

Model Details

Model Description

This DistilBERT model has been fine-tuned for sentiment analysis on the IMDb dataset. It is designed to be lightweight and efficient, making it suitable for deployment on low-end PCs and machines. The model can accurately classify movie reviews as positive or negative.

  • Developed by: Saif jabri
  • Model type: DistilBERT for Sequence Classification
  • Language(s) (NLP): English
  • License: Apache-2.0
  • Finetuned from model: distilbert-base-uncased

Model Sources

Uses

Direct Use

This model can be used directly for sentiment analysis on English text data, particularly movie reviews.

Downstream Use

The model can be fine-tuned further for other sentiment analysis tasks or integrated into larger applications requiring sentiment classification.

Out-of-Scope Use

This model is not suitable for non-English text or tasks unrelated to sentiment analysis.

Bias, Risks, and Limitations

While the model performs well on the IMDb dataset, it may have biases related to the data it was trained on. It might not generalize well to other domains or nuanced sentiment contexts.

Recommendations

Users should be aware of the model's limitations and biases. Testing the model on a variety of inputs is recommended to understand its behavior and performance across different scenarios.

How to Get Started with the Model

Use the code below to get started with the model:

from transformers import pipeline

classifier = pipeline("sentiment-analysis", model="saiffff/distilbert-imdb-sentiment")
result = classifier("This movie was fantastic! I loved every moment of it.")
print(result)