update model card README.md
Browse files
README.md
CHANGED
@@ -14,48 +14,43 @@ should probably proofread and complete it, then remove this comment. -->
|
|
14 |
|
15 |
This model is a fine-tuned version of [cointegrated/rubert-tiny2](https://huggingface.co/cointegrated/rubert-tiny2) on an unknown dataset.
|
16 |
It achieves the following results on the evaluation set:
|
17 |
-
-
|
18 |
-
-
|
19 |
-
-
|
20 |
-
-
|
21 |
-
-
|
22 |
-
-
|
23 |
-
-
|
24 |
-
-
|
25 |
-
-
|
26 |
-
-
|
27 |
-
-
|
28 |
-
-
|
29 |
-
-
|
30 |
-
-
|
31 |
-
-
|
32 |
-
-
|
33 |
-
-
|
34 |
-
-
|
35 |
-
-
|
36 |
-
-
|
37 |
-
-
|
38 |
-
-
|
39 |
-
-
|
40 |
-
-
|
41 |
-
-
|
42 |
-
-
|
43 |
-
-
|
44 |
-
-
|
45 |
-
-
|
46 |
-
-
|
47 |
-
-
|
48 |
-
-
|
49 |
-
-
|
50 |
-
-
|
51 |
-
-
|
52 |
-
-
|
53 |
-
-
|
54 |
-
- eval_runtime: 1.6021
|
55 |
-
- eval_samples_per_second: 355.786
|
56 |
-
- eval_steps_per_second: 355.786
|
57 |
-
- epoch: 1.0
|
58 |
-
- step: 4864
|
59 |
|
60 |
## Model description
|
61 |
|
@@ -74,16 +69,27 @@ More information needed
|
|
74 |
### Training hyperparameters
|
75 |
|
76 |
The following hyperparameters were used during training:
|
77 |
-
- learning_rate:
|
78 |
- train_batch_size: 1
|
79 |
- eval_batch_size: 1
|
80 |
-
- seed:
|
81 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
82 |
- lr_scheduler_type: linear
|
83 |
-
- lr_scheduler_warmup_ratio: 0.
|
84 |
- num_epochs: 5
|
85 |
- mixed_precision_training: Native AMP
|
86 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
87 |
### Framework versions
|
88 |
|
89 |
- Transformers 4.28.1
|
|
|
14 |
|
15 |
This model is a fine-tuned version of [cointegrated/rubert-tiny2](https://huggingface.co/cointegrated/rubert-tiny2) on an unknown dataset.
|
16 |
It achieves the following results on the evaluation set:
|
17 |
+
- Loss: 0.3357
|
18 |
+
- Benefactive Precision: 1.0
|
19 |
+
- Benefactive Recall: 0.5
|
20 |
+
- Benefactive F1: 0.6667
|
21 |
+
- Benefactive Number: 2
|
22 |
+
- Causator Precision: 1.0
|
23 |
+
- Causator Recall: 1.0
|
24 |
+
- Causator F1: 1.0
|
25 |
+
- Causator Number: 12
|
26 |
+
- Cause Precision: 0.4
|
27 |
+
- Cause Recall: 0.4
|
28 |
+
- Cause F1: 0.4000
|
29 |
+
- Cause Number: 5
|
30 |
+
- Contrsubject Precision: 0.75
|
31 |
+
- Contrsubject Recall: 0.6667
|
32 |
+
- Contrsubject F1: 0.7059
|
33 |
+
- Contrsubject Number: 9
|
34 |
+
- Deliberative Precision: 1.0
|
35 |
+
- Deliberative Recall: 1.0
|
36 |
+
- Deliberative F1: 1.0
|
37 |
+
- Deliberative Number: 4
|
38 |
+
- Experiencer Precision: 0.7442
|
39 |
+
- Experiencer Recall: 0.8101
|
40 |
+
- Experiencer F1: 0.7758
|
41 |
+
- Experiencer Number: 79
|
42 |
+
- Object Precision: 0.7551
|
43 |
+
- Object Recall: 0.7450
|
44 |
+
- Object F1: 0.7500
|
45 |
+
- Object Number: 149
|
46 |
+
- Predicate Precision: 0.9809
|
47 |
+
- Predicate Recall: 0.9885
|
48 |
+
- Predicate F1: 0.9847
|
49 |
+
- Predicate Number: 260
|
50 |
+
- Overall Precision: 0.8705
|
51 |
+
- Overall Recall: 0.8788
|
52 |
+
- Overall F1: 0.8746
|
53 |
+
- Overall Accuracy: 0.9411
|
|
|
|
|
|
|
|
|
|
|
54 |
|
55 |
## Model description
|
56 |
|
|
|
69 |
### Training hyperparameters
|
70 |
|
71 |
The following hyperparameters were used during training:
|
72 |
+
- learning_rate: 0.00010372880304918982
|
73 |
- train_batch_size: 1
|
74 |
- eval_batch_size: 1
|
75 |
+
- seed: 923789
|
76 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
77 |
- lr_scheduler_type: linear
|
78 |
+
- lr_scheduler_warmup_ratio: 0.29
|
79 |
- num_epochs: 5
|
80 |
- mixed_precision_training: Native AMP
|
81 |
|
82 |
+
### Training results
|
83 |
+
|
84 |
+
| Training Loss | Epoch | Step | Validation Loss | Benefactive Precision | Benefactive Recall | Benefactive F1 | Benefactive Number | Causator Precision | Causator Recall | Causator F1 | Causator Number | Cause Precision | Cause Recall | Cause F1 | Cause Number | Contrsubject Precision | Contrsubject Recall | Contrsubject F1 | Contrsubject Number | Deliberative Precision | Deliberative Recall | Deliberative F1 | Deliberative Number | Experiencer Precision | Experiencer Recall | Experiencer F1 | Experiencer Number | Object Precision | Object Recall | Object F1 | Object Number | Predicate Precision | Predicate Recall | Predicate F1 | Predicate Number | Overall Precision | Overall Recall | Overall F1 | Overall Accuracy |
|
85 |
+
|:-------------:|:-----:|:-----:|:---------------:|:---------------------:|:------------------:|:--------------:|:------------------:|:------------------:|:---------------:|:-----------:|:---------------:|:---------------:|:------------:|:--------:|:------------:|:----------------------:|:-------------------:|:---------------:|:-------------------:|:----------------------:|:-------------------:|:---------------:|:-------------------:|:---------------------:|:------------------:|:--------------:|:------------------:|:----------------:|:-------------:|:---------:|:-------------:|:-------------------:|:----------------:|:------------:|:----------------:|:-----------------:|:--------------:|:----------:|:----------------:|
|
86 |
+
| 0.3041 | 1.0 | 4864 | 0.3394 | 0.0 | 0.0 | 0.0 | 2 | 1.0 | 0.4167 | 0.5882 | 12 | 0.0 | 0.0 | 0.0 | 5 | 1.0 | 0.1111 | 0.2000 | 9 | 0.0 | 0.0 | 0.0 | 4 | 0.8372 | 0.4557 | 0.5902 | 79 | 0.8730 | 0.3691 | 0.5189 | 149 | 0.9884 | 0.9846 | 0.9865 | 260 | 0.9515 | 0.6788 | 0.7924 | 0.9141 |
|
87 |
+
| 0.3178 | 2.0 | 9728 | 0.2692 | 0.0 | 0.0 | 0.0 | 2 | 1.0 | 1.0 | 1.0 | 12 | 1.0 | 0.2 | 0.3333 | 5 | 1.0 | 0.2222 | 0.3636 | 9 | 1.0 | 0.5 | 0.6667 | 4 | 0.7403 | 0.7215 | 0.7308 | 79 | 0.8523 | 0.5034 | 0.6329 | 149 | 0.9808 | 0.9846 | 0.9827 | 260 | 0.9142 | 0.7788 | 0.8411 | 0.9321 |
|
88 |
+
| 0.124 | 3.0 | 14592 | 0.2990 | 1.0 | 0.5 | 0.6667 | 2 | 1.0 | 1.0 | 1.0 | 12 | 0.0 | 0.0 | 0.0 | 5 | 0.75 | 0.6667 | 0.7059 | 9 | 1.0 | 0.5 | 0.6667 | 4 | 0.7386 | 0.8228 | 0.7784 | 79 | 0.8 | 0.6980 | 0.7455 | 149 | 0.9885 | 0.9885 | 0.9885 | 260 | 0.8904 | 0.8596 | 0.8748 | 0.9435 |
|
89 |
+
| 0.104 | 4.0 | 19456 | 0.2852 | 1.0 | 0.5 | 0.6667 | 2 | 0.9231 | 1.0 | 0.9600 | 12 | 0.4286 | 0.6 | 0.5 | 5 | 0.6 | 0.6667 | 0.6316 | 9 | 1.0 | 0.75 | 0.8571 | 4 | 0.7253 | 0.8354 | 0.7765 | 79 | 0.7044 | 0.7517 | 0.7273 | 149 | 0.9847 | 0.9885 | 0.9866 | 260 | 0.8440 | 0.8846 | 0.8638 | 0.9359 |
|
90 |
+
| 0.0918 | 5.0 | 24320 | 0.3357 | 1.0 | 0.5 | 0.6667 | 2 | 1.0 | 1.0 | 1.0 | 12 | 0.4 | 0.4 | 0.4000 | 5 | 0.75 | 0.6667 | 0.7059 | 9 | 1.0 | 1.0 | 1.0 | 4 | 0.7442 | 0.8101 | 0.7758 | 79 | 0.7551 | 0.7450 | 0.7500 | 149 | 0.9809 | 0.9885 | 0.9847 | 260 | 0.8705 | 0.8788 | 0.8746 | 0.9411 |
|
91 |
+
|
92 |
+
|
93 |
### Framework versions
|
94 |
|
95 |
- Transformers 4.28.1
|