Update README.md
Browse files
README.md
CHANGED
|
@@ -9,6 +9,13 @@ tags:
|
|
| 9 |
- transformers
|
| 10 |
- bert
|
| 11 |
pipeline_tag: text-classification
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
# AG-News BERT Classification
|
|
@@ -149,5 +156,4 @@ print(prediction)
|
|
| 149 |
year = {2024},
|
| 150 |
publisher = {Hugging Face},
|
| 151 |
howpublished = {\url{https://huggingface.co/mansoorhamidzadeh/ag-news-bert-classification}},
|
| 152 |
-
}
|
| 153 |
-
|
|
|
|
| 9 |
- transformers
|
| 10 |
- bert
|
| 11 |
pipeline_tag: text-classification
|
| 12 |
+
datasets:
|
| 13 |
+
- fancyzhx/ag_news
|
| 14 |
+
metrics:
|
| 15 |
+
- accuracy
|
| 16 |
+
- recall
|
| 17 |
+
- precision
|
| 18 |
+
- f1
|
| 19 |
---
|
| 20 |
|
| 21 |
# AG-News BERT Classification
|
|
|
|
| 156 |
year = {2024},
|
| 157 |
publisher = {Hugging Face},
|
| 158 |
howpublished = {\url{https://huggingface.co/mansoorhamidzadeh/ag-news-bert-classification}},
|
| 159 |
+
}
|
|
|