SMG0 commited on
Commit
cacebed
·
1 Parent(s): 041fb61

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -1
README.md CHANGED
@@ -2,6 +2,9 @@
2
  base_model: UBC-NLP/MARBERTv2
3
  tags:
4
  - generated_from_trainer
 
 
 
5
  model-index:
6
  - name: Model3_Marabertv2_T1_WOS
7
  results: []
@@ -12,7 +15,13 @@ should probably proofread and complete it, then remove this comment. -->
12
 
13
  # Model3_Marabertv2_T1_WOS
14
 
15
- This model is a fine-tuned version of [UBC-NLP/MARBERTv2](https://huggingface.co/UBC-NLP/MARBERTv2) on an unknown dataset.
 
 
 
 
 
 
16
 
17
  ## Model description
18
 
@@ -39,6 +48,17 @@ The following hyperparameters were used during training:
39
  - lr_scheduler_type: linear
40
  - num_epochs: 15
41
 
 
 
 
 
 
 
 
 
 
 
 
42
  ### Framework versions
43
 
44
  - Transformers 4.31.0
 
2
  base_model: UBC-NLP/MARBERTv2
3
  tags:
4
  - generated_from_trainer
5
+ metrics:
6
+ - f1
7
+ - accuracy
8
  model-index:
9
  - name: Model3_Marabertv2_T1_WOS
10
  results: []
 
15
 
16
  # Model3_Marabertv2_T1_WOS
17
 
18
+ This model is a fine-tuned version of [UBC-NLP/MARBERTv2](https://huggingface.co/UBC-NLP/MARBERTv2) on the None dataset.
19
+ It achieves the following results on the evaluation set:
20
+ - Loss: 0.2308
21
+ - F1: 0.8430
22
+ - F1 Macro: 0.7804
23
+ - Roc Auc: 0.9048
24
+ - Accuracy: 0.8142
25
 
26
  ## Model description
27
 
 
48
  - lr_scheduler_type: linear
49
  - num_epochs: 15
50
 
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss | F1 | F1 Macro | Roc Auc | Accuracy |
54
+ |:-------------:|:-----:|:----:|:---------------:|:------:|:--------:|:-------:|:--------:|
55
+ | 0.2194 | 1.0 | 507 | 0.1556 | 0.8330 | 0.7507 | 0.8909 | 0.7947 |
56
+ | 0.1166 | 2.0 | 1014 | 0.1850 | 0.8269 | 0.7439 | 0.8920 | 0.8010 |
57
+ | 0.0747 | 3.0 | 1521 | 0.1915 | 0.8368 | 0.7724 | 0.8992 | 0.8115 |
58
+ | 0.0445 | 4.0 | 2028 | 0.2034 | 0.8398 | 0.7695 | 0.9014 | 0.8149 |
59
+ | 0.0301 | 5.0 | 2535 | 0.2308 | 0.8430 | 0.7804 | 0.9048 | 0.8142 |
60
+
61
+
62
  ### Framework versions
63
 
64
  - Transformers 4.31.0