Text Classification
Transformers
PyTorch
roberta
topic
classification
news
Eval Results (legacy)
text-embeddings-inference
Instructions to use dstefa/roberta-base_topic_classification_nyt_news with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use dstefa/roberta-base_topic_classification_nyt_news with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="dstefa/roberta-base_topic_classification_nyt_news")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("dstefa/roberta-base_topic_classification_nyt_news") model = AutoModelForSequenceClassification.from_pretrained("dstefa/roberta-base_topic_classification_nyt_news", device_map="auto") - Inference
- Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -50,7 +50,7 @@ pipeline_tag: text-classification
|
|
| 50 |
|
| 51 |
# roberta-base_topic_classification_nyt_news
|
| 52 |
|
| 53 |
-
This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on the NYT News dataset (https://www.kaggle.com/datasets/aryansingh0909/nyt-articles-21m-2000-present).
|
| 54 |
It achieves the following results on the test set of 51200 cases:
|
| 55 |
- Accuracy: 0.91
|
| 56 |
- F1: 0.91
|
|
|
|
| 50 |
|
| 51 |
# roberta-base_topic_classification_nyt_news
|
| 52 |
|
| 53 |
+
This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on the NYT News dataset, which contains 256,000 news titles from articles published from 2000 to the present (https://www.kaggle.com/datasets/aryansingh0909/nyt-articles-21m-2000-present).
|
| 54 |
It achieves the following results on the test set of 51200 cases:
|
| 55 |
- Accuracy: 0.91
|
| 56 |
- F1: 0.91
|