Hugging Face's logo

pabRomero
/
BioMedRoBERTa-full-finetuned-ner-pablo

Token Classification
Transformers
TensorBoard
ONNX
Safetensors
roberta
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
pabRomero commited on
Commit
71b887d
•
1 Parent(s): e17315f

Training complete

Browse files
Files changed (1) hide show
  1. README.md +15 -15
README.md CHANGED
@@ -20,11 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
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.1094
24
- - Precision: 0.8151
25
- - Recall: 0.7999
26
- - F1: 0.8074
27
- - Accuracy: 0.9737
28
 
29
  ## Model description
30
 
@@ -43,9 +43,9 @@ More information needed
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
@@ -55,13 +55,13 @@ The following hyperparameters were used during training:
55
 
56
  ### Training results
57
 
58
- | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
59
- |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
60
- | 0.3008 | 1.0 | 613 | 0.0937 | 0.8042 | 0.7705 | 0.7870 | 0.9727 |
61
- | 0.0799 | 2.0 | 1226 | 0.0873 | 0.8103 | 0.7936 | 0.8019 | 0.9744 |
62
- | 0.0649 | 3.0 | 1839 | 0.0888 | 0.8179 | 0.7930 | 0.8053 | 0.9748 |
63
- | 0.053 | 4.0 | 2452 | 0.0966 | 0.8082 | 0.7978 | 0.8029 | 0.9732 |
64
- | 0.0321 | 5.0 | 3065 | 0.1094 | 0.8151 | 0.7999 | 0.8074 | 0.9737 |
65
 
66
 
67
  ### Framework versions
 
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.0985
24
+ - Precision: 0.8190
25
+ - Recall: 0.8095
26
+ - F1: 0.8142
27
+ - Accuracy: 0.9779
28
 
29
  ## Model description
30
 
 
43
  ### Training hyperparameters
44
 
45
  The following hyperparameters were used during training:
46
+ - learning_rate: 3e-05
47
+ - train_batch_size: 2
48
+ - eval_batch_size: 2
49
  - seed: 42
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: linear
 
55
 
56
  ### Training results
57
 
58
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
59
+ |:-------------:|:-----:|:-----:|:---------------:|:---------:|:------:|:------:|:--------:|
60
+ | 0.0864 | 1.0 | 2509 | 0.0915 | 0.8408 | 0.7476 | 0.7915 | 0.9753 |
61
+ | 0.0616 | 2.0 | 5018 | 0.0766 | 0.8116 | 0.7911 | 0.8012 | 0.9782 |
62
+ | 0.0439 | 3.0 | 7527 | 0.0838 | 0.8132 | 0.8024 | 0.8078 | 0.9776 |
63
+ | 0.0338 | 4.0 | 10036 | 0.0891 | 0.8207 | 0.8030 | 0.8118 | 0.9775 |
64
+ | 0.0268 | 5.0 | 12545 | 0.0985 | 0.8190 | 0.8095 | 0.8142 | 0.9779 |
65
 
66
 
67
  ### Framework versions