metadata

license: mit
pipeline_tag: text-classification

Fine-tuned RoBERTa for Sentiment Analysis on Amazon Reviews

This is a fine-tuned version of cardiffnlp/twitter-roberta-base-sentiment-latest on the Amazon Reviews dataset for sentiment analysis.

Model Details

Model Name: AnkitAI/reviews-roberta-base-sentiment-analysis
Base Model: cardiffnlp/twitter-roberta-base-sentiment-latest
Dataset: Amazon Reviews
Fine-tuning: This model was fine-tuned for sentiment analysis with a classification head for binary sentiment classification (positive and negative).

Training

The model was trained using the following parameters:

Learning Rate: 2e-5
Batch Size: 16
Epochs: 3
Weight Decay: 0.01
Evaluation Strategy: Epoch

Usage

You can use this model directly with the Hugging Face transformers library:

from transformers import RobertaForSequenceClassification, RobertaTokenizer

model_name = "AnkitAI/reviews-roberta-base-sentiment-analysis"
model = RobertaForSequenceClassification.from_pretrained(model_name)
tokenizer = RobertaTokenizer.from_pretrained(model_name)

# Example usage
inputs = tokenizer("This product is great!", return_tensors="pt")
outputs = model(**inputs)

License

This model is licensed under the mit license