Hugging Face's logo

pabRomero
/
RoBERTa-full-finetuned-ner-pablo

Token Classification
Transformers
TensorBoard
Safetensors
roberta
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
pabRomero commited on
Commit
cce7bc4
•
1 Parent(s): 2f3428e

Training complete

Browse files
Files changed (3) hide show
  1. README.md +15 -20
  2. model.safetensors +1 -1
  3. runs/Sep05_12-17-25_83295d15965e/events.out.tfevents.1725538646.83295d15965e.5325.6 +2 -2
README.md CHANGED
@@ -21,11 +21,11 @@ should probably proofread and complete it, then remove this comment. -->
21
 
22
  This model is a fine-tuned version of [FacebookAI/roberta-base](https://huggingface.co/FacebookAI/roberta-base) on the None dataset.
23
  It achieves the following results on the evaluation set:
24
- - Loss: 0.0908
25
- - Precision: 0.815
26
- - Recall: 0.8050
27
- - F1: 0.8100
28
- - Accuracy: 0.9767
29
 
30
  ## Model description
31
 
@@ -44,32 +44,27 @@ More information needed
44
  ### Training hyperparameters
45
 
46
  The following hyperparameters were used during training:
47
- - learning_rate: 0.0002
48
- - train_batch_size: 16
49
- - eval_batch_size: 16
50
  - seed: 42
51
- - gradient_accumulation_steps: 4
52
- - total_train_batch_size: 64
53
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
54
  - lr_scheduler_type: linear
55
- - lr_scheduler_warmup_ratio: 0.05
56
- - num_epochs: 5
57
  - mixed_precision_training: Native AMP
58
 
59
  ### Training results
60
 
61
- | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
62
- |:-------------:|:------:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
63
- | No log | 0.9970 | 252 | 0.1094 | 0.6862 | 0.7348 | 0.7097 | 0.9688 |
64
- | 0.1923 | 1.9980 | 505 | 0.0836 | 0.8005 | 0.7868 | 0.7936 | 0.9762 |
65
- | 0.1923 | 2.9990 | 758 | 0.0816 | 0.8109 | 0.7855 | 0.7980 | 0.9755 |
66
- | 0.0504 | 4.0 | 1011 | 0.0839 | 0.8073 | 0.8028 | 0.8051 | 0.9763 |
67
- | 0.0504 | 4.9852 | 1260 | 0.0908 | 0.815 | 0.8050 | 0.8100 | 0.9767 |
68
 
69
 
70
  ### Framework versions
71
 
72
  - Transformers 4.44.2
73
- - Pytorch 2.4.0+cu121
74
  - Datasets 2.21.0
75
  - Tokenizers 0.19.1
 
21
 
22
  This model is a fine-tuned version of [FacebookAI/roberta-base](https://huggingface.co/FacebookAI/roberta-base) on the None dataset.
23
  It achieves the following results on the evaluation set:
24
+ - Loss: 0.0616
25
+ - Precision: 0.8439
26
+ - Recall: 0.8346
27
+ - F1: 0.8392
28
+ - Accuracy: 0.9811
29
 
30
  ## Model description
31
 
 
44
  ### Training hyperparameters
45
 
46
  The following hyperparameters were used during training:
47
+ - learning_rate: 5e-05
48
+ - train_batch_size: 32
49
+ - eval_batch_size: 32
50
  - seed: 42
 
 
51
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
52
  - lr_scheduler_type: linear
53
+ - lr_scheduler_warmup_ratio: 0.1
54
+ - num_epochs: 2
55
  - mixed_precision_training: Native AMP
56
 
57
  ### Training results
58
 
59
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
60
+ |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
61
+ | No log | 1.0 | 231 | 0.0805 | 0.7890 | 0.8039 | 0.7964 | 0.9755 |
62
+ | No log | 2.0 | 462 | 0.0616 | 0.8439 | 0.8346 | 0.8392 | 0.9811 |
 
 
 
63
 
64
 
65
  ### Framework versions
66
 
67
  - Transformers 4.44.2
68
+ - Pytorch 2.4.1+cu121
69
  - Datasets 2.21.0
70
  - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9ad5eb8cdfdf6e9c7724ce664864d8b85d469237b11a02ba214d66b94bd8919b
3
  size 496302532
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0f75650dd2452bfca47d17cf7e83a3d70c403f6b05eba17afe9a9a46b03de411
3
  size 496302532
runs/Sep05_12-17-25_83295d15965e/events.out.tfevents.1725538646.83295d15965e.5325.6 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4da21844b916f7907affc0044db57d9802b6b1a1799864e51b4b53f443360dca
3
- size 6234
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5fd54d2493d5eb313ae9ee04a53fa53df45c3ef4410321c75d341572d8a134e9
3
+ size 7060