pabRomero commited on
Commit
84046bf
1 Parent(s): aaa20fc

Training complete

Browse files
README.md CHANGED
@@ -1,73 +1,72 @@
1
- ---
2
- base_model: allenai/biomed_roberta_base
3
- tags:
4
- - generated_from_trainer
5
- metrics:
6
- - precision
7
- - recall
8
- - f1
9
- - accuracy
10
- model-index:
11
- - name: BioMedRoBERTa-full-finetuned-ner-pablo
12
- results: []
13
- ---
14
-
15
- <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
- should probably proofread and complete it, then remove this comment. -->
17
-
18
- # BioMedRoBERTa-full-finetuned-ner-pablo
19
-
20
- This model is a fine-tuned version of [allenai/biomed_roberta_base](https://huggingface.co/allenai/biomed_roberta_base) on the None dataset.
21
- It achieves the following results on the evaluation set:
22
- - Loss: 0.1128
23
- - Precision: 0.8205
24
- - Recall: 0.8113
25
- - F1: 0.8159
26
- - Accuracy: 0.9729
27
-
28
- ## Model description
29
-
30
- More information needed
31
-
32
- ## Intended uses & limitations
33
-
34
- More information needed
35
-
36
- ## Training and evaluation data
37
-
38
- More information needed
39
-
40
- ## Training procedure
41
-
42
- ### Training hyperparameters
43
-
44
- The following hyperparameters were used during training:
45
- - learning_rate: 0.0002
46
- - train_batch_size: 16
47
- - eval_batch_size: 16
48
- - seed: 42
49
- - gradient_accumulation_steps: 4
50
- - total_train_batch_size: 64
51
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
52
- - lr_scheduler_type: linear
53
- - lr_scheduler_warmup_ratio: 0.05
54
- - num_epochs: 5
55
- - mixed_precision_training: Native AMP
56
-
57
- ### Training results
58
-
59
- | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
60
- |:-------------:|:------:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
61
- | 0.2807 | 0.9996 | 652 | 0.1084 | 0.8072 | 0.7593 | 0.7825 | 0.9703 |
62
- | 0.099 | 1.9992 | 1304 | 0.0994 | 0.7885 | 0.7819 | 0.7852 | 0.9704 |
63
- | 0.0781 | 2.9989 | 1956 | 0.0902 | 0.8206 | 0.7904 | 0.8052 | 0.9727 |
64
- | 0.049 | 4.0 | 2609 | 0.0997 | 0.8223 | 0.8101 | 0.8161 | 0.9731 |
65
- | 0.0386 | 4.9981 | 3260 | 0.1128 | 0.8205 | 0.8113 | 0.8159 | 0.9729 |
66
-
67
-
68
- ### Framework versions
69
-
70
- - Transformers 4.44.0
71
- - Pytorch 2.4.0+cu124
72
- - Datasets 2.21.0
73
- - Tokenizers 0.19.1
 
1
+ ---
2
+ library_name: transformers
3
+ base_model: allenai/biomed_roberta_base
4
+ tags:
5
+ - generated_from_trainer
6
+ metrics:
7
+ - precision
8
+ - recall
9
+ - f1
10
+ - accuracy
11
+ model-index:
12
+ - name: BioMedRoBERTa-full-finetuned-ner-pablo
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # BioMedRoBERTa-full-finetuned-ner-pablo
20
+
21
+ This model is a fine-tuned version of [allenai/biomed_roberta_base](https://huggingface.co/allenai/biomed_roberta_base) on the None dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 0.0957
24
+ - Precision: 0.8174
25
+ - Recall: 0.8204
26
+ - F1: 0.8189
27
+ - Accuracy: 0.9769
28
+
29
+ ## Model description
30
+
31
+ More information needed
32
+
33
+ ## Intended uses & limitations
34
+
35
+ More information needed
36
+
37
+ ## Training and evaluation data
38
+
39
+ More information needed
40
+
41
+ ## Training procedure
42
+
43
+ ### Training hyperparameters
44
+
45
+ The following hyperparameters were used during training:
46
+ - learning_rate: 0.0001
47
+ - train_batch_size: 128
48
+ - eval_batch_size: 128
49
+ - seed: 42
50
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
+ - lr_scheduler_type: linear
52
+ - lr_scheduler_warmup_ratio: 0.05
53
+ - num_epochs: 5
54
+ - mixed_precision_training: Native AMP
55
+
56
+ ### Training results
57
+
58
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
59
+ |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
60
+ | No log | 1.0 | 427 | 0.0818 | 0.8084 | 0.7885 | 0.7983 | 0.9756 |
61
+ | 0.263 | 2.0 | 854 | 0.0760 | 0.8029 | 0.8081 | 0.8055 | 0.9766 |
62
+ | 0.0608 | 3.0 | 1281 | 0.0818 | 0.7963 | 0.8199 | 0.8079 | 0.9754 |
63
+ | 0.046 | 4.0 | 1708 | 0.0904 | 0.8048 | 0.8232 | 0.8139 | 0.9759 |
64
+ | 0.0327 | 5.0 | 2135 | 0.0957 | 0.8174 | 0.8204 | 0.8189 | 0.9769 |
65
+
66
+
67
+ ### Framework versions
68
+
69
+ - Transformers 4.44.1
70
+ - Pytorch 2.4.0+cu121
71
+ - Datasets 2.21.0
72
+ - Tokenizers 0.19.1
 
runs/Aug22_08-42-19_26b01926c00f/events.out.tfevents.1724316140.26b01926c00f.1526.2 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fbd5910b252c605d41520522c68802cc3cd453c90f2a22c5dd31e725bd55c76d
3
- size 8466
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8905bc5ad6b0811bec756d0ea444b2e654f0f6383f6ced9e89df2d486a83fb2
3
+ size 9292