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

bca3dac

·

1 Parent(s): fd5413b

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: 0.8083
-- Accuracy: 0.7698
-- F1 Macro: 0.5907
 ## Model description
@@ -52,16 +52,16 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Macro |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|
-| No log        | 0.98  | 26   | 1.3068          | 0.5175   | 0.1137   |
-| No log        | 2.0   | 53   | 1.1049          | 0.6053   | 0.2234   |
-| No log        | 2.98  | 79   | 0.9886          | 0.6930   | 0.3290   |
-| 1.2122        | 4.0   | 106  | 0.9229          | 0.7018   | 0.3507   |
-| 1.2122        | 4.98  | 132  | 0.8905          | 0.7018   | 0.3546   |
-| 1.2122        | 6.0   | 159  | 0.8987          | 0.7105   | 0.3595   |
-| 1.2122        | 6.98  | 185  | 0.8556          | 0.7018   | 0.4048   |
-| 0.6319        | 8.0   | 212  | 0.8385          | 0.7368   | 0.4711   |
-| 0.6319        | 8.98  | 238  | 0.8320          | 0.7281   | 0.4660   |
-| 0.6319        | 9.81  | 260  | 0.8227          | 0.7281   | 0.4660   |
 ### 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.8592
+- Accuracy: 0.7194
+- F1 Macro: 0.4819
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 Macro |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|
+| No log        | 0.98  | 26   | 1.3259          | 0.5175   | 0.1137   |
+| No log        | 2.0   | 53   | 1.1130          | 0.6228   | 0.2479   |
+| No log        | 2.98  | 79   | 1.0243          | 0.6667   | 0.3126   |
+| 1.2194        | 4.0   | 106  | 0.9297          | 0.7018   | 0.3506   |
+| 1.2194        | 4.98  | 132  | 0.9334          | 0.7018   | 0.3593   |
+| 1.2194        | 6.0   | 159  | 0.8904          | 0.7368   | 0.5001   |
+| 1.2194        | 6.98  | 185  | 0.8714          | 0.7281   | 0.4661   |
+| 0.6526        | 8.0   | 212  | 0.8810          | 0.7368   | 0.4847   |
+| 0.6526        | 8.98  | 238  | 0.8807          | 0.7456   | 0.5552   |
+| 0.6526        | 9.81  | 260  | 0.8945          | 0.7193   | 0.5422   |
 ### Framework versions