Downloads · 30 days
0
final-sentiment-model-go-raw
Go-Raw/final-sentiment-model-go-raw
final-sentiment-model-go-raw is a text classification model from Go-Raw. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
This is a fine-tuned roberta-base model for multi-class sentiment classification. It was trained on a custom dataset of ~240k examples with 3 sentiment classes: - 0: Negative - 1: Positive - 2: Neutral
Access
Public
Updated Jul 7, 2025
Repo size
—
Likes
1
Public
Repo makeup
Click a slice to open those files.
.md1.9 KB · 55%
At a glance
- Task
- Text Classification
- License
- apache-2.0
- Access
- Public
- Created
- Jul 6, 2025
- Updated
- Jul 7, 2025
- SHA
- 21927784
Try a prompt
Base models
Datasets
Model card
From the Hugging Face model README
Final Sentiment Model - Go-Raw
Model description
This is a fine-tuned roberta-base model for multi-class sentiment classification.
It was trained on a custom dataset of ~240k examples with 3 sentiment classes:
- 0: Negative
- 1: Positive
- 2: Neutral
The model shows significant improvement over the base model on this task.
Intended uses & limitations
- ✅ Suitable for English text sentiment analysis.
- 🚫 Not tested on other languages or domains beyond training data.
- 🚫 Not suitable for detecting abusive, toxic, or hate speech.
Training details
- Base model:
roberta-base - Epochs: 3
- Learning rate: 2e-5
- Batch size: 8
- Optimizer: AdamW
Evaluation
Dataset
- Train set: 1,94,038 examples
- Test set: 48,510 examples
Performance
| Metric | Base Model | Fine-tuned Model |
|---|---|---|
| Accuracy | 34.1% | 88.1% |
| Macro F1 | 24.3% | 87.5% |
| Weighted F1 | 27.1% | 88.1% |
Per-class metrics
| Class | Precision | Recall | F1-score |
|---|---|---|---|
| 0 (Negative) | 85.3% | 83.1% | 84.2% |
| 1 (Neutral) | 91.4% | 89.8% | 90.5% |
| 2 (Positive) | 86.0% | 89.4% | 87.7% |
How to use
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("Go-Raw/final-sentiment-model-go-raw")
tokenizer = AutoTokenizer.from_pretrained("Go-Raw/final-sentiment-model-go-raw")
text = "I absolutely love this!"
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)
predicted_class = outputs.logits.argmax().item()
print(predicted_class)
- Task
- Text Classification
- License
- apache-2.0
- Languages
- en
- Created
- Jul 6, 2025
- Updated
- Jul 7, 2025