ltuzova
/

citation_intent_classification_roberta

Text Classification

Generated from Trainer

text-embeddings-inference

Model card Files Files and versions

Metrics Training metrics Community

ltuzova commited on Apr 20, 2024

Commit

78372bb

·

1 Parent(s): 6c1486f

update model card README.md

Files changed (1) hide show

README.md +13 -13

README.md CHANGED Viewed

@@ -16,9 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.0731
-- Accuracy: 0.7842
-- F1 Macro: 0.6927
 ## Model description
@@ -52,16 +52,16 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Macro |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|
-| 1.3931        | 1.0   | 105  | 1.0457          | 0.6842   | 0.3071   |
-| 1.0487        | 2.0   | 211  | 0.8936          | 0.6842   | 0.3201   |
-| 0.8369        | 3.0   | 316  | 0.7841          | 0.7281   | 0.4126   |
-| 0.5949        | 4.0   | 422  | 0.6737          | 0.7807   | 0.6027   |
-| 0.431         | 5.0   | 527  | 0.7379          | 0.7632   | 0.6569   |
-| 0.3456        | 6.0   | 633  | 0.7826          | 0.7719   | 0.6541   |
-| 0.2285        | 7.0   | 738  | 0.7410          | 0.8158   | 0.7300   |
-| 0.1757        | 8.0   | 844  | 0.8297          | 0.7982   | 0.7306   |
-| 0.1577        | 9.0   | 949  | 0.8780          | 0.7982   | 0.7212   |
-| 0.0973        | 9.95  | 1050 | 0.9575          | 0.8158   | 0.7410   |
 ### Framework versions

 This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on the None dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.9427
+- Accuracy: 0.7986
+- F1 Macro: 0.6913
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Macro |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|
+| 1.4059        | 1.0   | 105  | 1.0585          | 0.6491   | 0.2714   |
+| 1.0405        | 2.0   | 211  | 0.9258          | 0.6842   | 0.3254   |
+| 0.8144        | 3.0   | 316  | 0.7686          | 0.7281   | 0.4907   |
+| 0.5887        | 4.0   | 422  | 0.7906          | 0.7456   | 0.5518   |
+| 0.404         | 5.0   | 527  | 0.6946          | 0.7719   | 0.7045   |
+| 0.3302        | 6.0   | 633  | 0.8840          | 0.7719   | 0.6330   |
+| 0.225         | 7.0   | 738  | 0.8200          | 0.8070   | 0.7258   |
+| 0.1843        | 8.0   | 844  | 0.8336          | 0.8070   | 0.7533   |
+| 0.16          | 9.0   | 949  | 0.8221          | 0.8246   | 0.7641   |
+| 0.0977        | 9.95  | 1050 | 0.8798          | 0.8246   | 0.7649   |
 ### Framework versions