Instructions to use textattack/roberta-base-ag-news with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use textattack/roberta-base-ag-news with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="textattack/roberta-base-ag-news")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("textattack/roberta-base-ag-news") model = AutoModelForSequenceClassification.from_pretrained("textattack/roberta-base-ag-news") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,5 +1,4 @@
|
|
| 1 |
-
## TextAttack Model
|
| 2 |
-
This `roberta-base` model was fine-tuned for sequence classification using TextAttack
|
| 3 |
and the ag_news dataset loaded using the `nlp` library. The model was fine-tuned
|
| 4 |
for 5 epochs with a batch size of 16, a learning
|
| 5 |
rate of 5e-05, and a maximum sequence length of 128.
|
|
|
|
| 1 |
+
## TextAttack Model CardThis `roberta-base` model was fine-tuned for sequence classification using TextAttack
|
|
|
|
| 2 |
and the ag_news dataset loaded using the `nlp` library. The model was fine-tuned
|
| 3 |
for 5 epochs with a batch size of 16, a learning
|
| 4 |
rate of 5e-05, and a maximum sequence length of 128.
|