Fine-Tuned Sentiment Model
This model is fine-tuned for Sentiment Analysis task, the model classifies a customer ticket into 5-categories of sentiments, namely:
- "Strong Negative"
- "Mild Negative"
- "Neutral"
- "Mild Positive"
- "Strong Positive"
*Point To Note*: The Customers are from these specific industries only:
- Food
- Cars
- Pet Food
- Furniture
- Beauty
Model Details
- **Model Architecture**: This fine-tuned model was built on a pre-trained model, "IDEA-CCNL/Erlangshen-Roberta-110M-Sentiment"
- **Training Dataset**: The Dataset was generated using the model, "meta-llama/Llama-3.2-1B-Instruct"
Example Usage-
To use this model for Sentiment Analysis:
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
tokenizer = AutoTokenizer.from_pretrained("rohittamidapati11/fine_tuned_sentiment_model_rt2")
model = AutoModelForSequenceClassification.from_pretrained("rohittamidapati11/fine_tuned_sentiment_model_rt2")
# Example input
inputs = tokenizer("The food was a bit bland, but the portion sizes were generous. I was looking forward to trying it, but it didn't quite live up to my expectations.", return_tensors='pt')
outputs = model(**inputs)
predicted_class = torch.argmax(outputs.logits, dim = 1).item()
print("Predicted Sentiment:", predicted_class)
- Downloads last month
- 15
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.