Update README.md
Browse files
README.md
CHANGED
|
@@ -17,7 +17,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 17 |
|
| 18 |
# roberta-news-classifier
|
| 19 |
|
| 20 |
-
This model is a fine-tuned version of [russellc/roberta-news-classifier](https://huggingface.co/russellc/roberta-news-classifier) on the
|
| 21 |
It achieves the following results on the evaluation set:
|
| 22 |
- Loss: 0.1043
|
| 23 |
- Accuracy: 0.9786
|
|
@@ -61,6 +61,31 @@ The following hyperparameters were used during training:
|
|
| 61 |
| 0.0825 | 4.0 | 492 | 0.1271 | 0.9714 | 0.9714 | 0.9714 | 0.9714 |
|
| 62 |
| 0.055 | 5.0 | 615 | 0.1296 | 0.9724 | 0.9724 | 0.9724 | 0.9724 |
|
| 63 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 64 |
|
| 65 |
### Framework versions
|
| 66 |
|
|
|
|
| 17 |
|
| 18 |
# roberta-news-classifier
|
| 19 |
|
| 20 |
+
This model is a fine-tuned version of [russellc/roberta-news-classifier](https://huggingface.co/russellc/roberta-news-classifier) on the custom(Kaggle) dataset.
|
| 21 |
It achieves the following results on the evaluation set:
|
| 22 |
- Loss: 0.1043
|
| 23 |
- Accuracy: 0.9786
|
|
|
|
| 61 |
| 0.0825 | 4.0 | 492 | 0.1271 | 0.9714 | 0.9714 | 0.9714 | 0.9714 |
|
| 62 |
| 0.055 | 5.0 | 615 | 0.1296 | 0.9724 | 0.9724 | 0.9724 | 0.9724 |
|
| 63 |
|
| 64 |
+
### Evaluation results
|
| 65 |
+
|
| 66 |
+
***** Running Prediction *****
|
| 67 |
+
Num examples = 980
|
| 68 |
+
Batch size = 64
|
| 69 |
+
|
| 70 |
+
precision recall f1-score support
|
| 71 |
+
|
| 72 |
+
dunya 0.99 0.96 0.97 147
|
| 73 |
+
ekonomi 0.96 0.96 0.96 141
|
| 74 |
+
kultur 0.97 0.99 0.98 142
|
| 75 |
+
saglik 0.99 0.98 0.98 148
|
| 76 |
+
siyaset 0.98 0.98 0.98 134
|
| 77 |
+
spor 1.00 1.00 1.00 139
|
| 78 |
+
teknoloji 0.96 0.98 0.97 129
|
| 79 |
+
accuracy -- -- 0.98 980
|
| 80 |
+
macro avg 0.98 0.98 0.98 980
|
| 81 |
+
weighted avg 0.98 0.98 0.98 980
|
| 82 |
+
|
| 83 |
+
|
| 84 |
+
|
| 85 |
+
|
| 86 |
+
|
| 87 |
+
|
| 88 |
+
|
| 89 |
|
| 90 |
### Framework versions
|
| 91 |
|