SetFit/bbc-news
Viewer • Updated • 2.23k • 1.96k • 22
How to use nicolasacosta/roberta-base_bbc-news with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="nicolasacosta/roberta-base_bbc-news") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("nicolasacosta/roberta-base_bbc-news")
model = AutoModelForSequenceClassification.from_pretrained("nicolasacosta/roberta-base_bbc-news")This model is a fine-tuned version of roberta-base trained on the bbc news dataset.:
It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Precision | Recall |
|---|---|---|---|---|---|---|---|
| 0.1976 | 1.0 | 154 | 0.1557 | 0.956 | 0.9563 | 0.9589 | 0.956 |
| 0.2669 | 2.0 | 308 | 0.1486 | 0.968 | 0.9679 | 0.9684 | 0.968 |
| 0.0181 | 3.0 | 462 | 0.2332 | 0.97 | 0.9700 | 0.9705 | 0.97 |
| 0.0957 | 4.0 | 616 | 0.1378 | 0.978 | 0.9781 | 0.9785 | 0.978 |
| 0.0009 | 5.0 | 770 | 0.1826 | 0.974 | 0.9740 | 0.9743 | 0.974 |
Base model
FacebookAI/roberta-base