CoolFace
Modelpublic

proc015/sentiment-model

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
0likes8downloads
Model Card

Sentiment Analysis Model

This model is a fine-tuned version of distilbert-base-uncased on the IMDb dataset for sentiment analysis.

Model Details

  • —Base model: distilbert-base-uncased
  • —Fine-tuning dataset: IMDb
  • —Task: Sentiment analysis (text classification)

Intended Use

The model is designed to classify text into positive or negative sentiment. You can use it for tasks such as:

  • —Analyzing product reviews.
  • —Social media sentiment analysis.
  • —General text classification tasks involving sentiment.

Limitations

  • —The model is fine-tuned on the IMDb dataset and may not generalize well to all domains or datasets.
  • —It may inherit biases from the IMDb dataset.

Example Usage

python
from transformers import pipeline

# Load the model
model_pipeline = pipeline("text-classification", model="proc015/sentiment-model")

# Run sentiment analysis
result = model_pipeline("I love this product!")
print(result)