Hugging Face's logo

dtorber
/
bertweet-base

Text Classification
Transformers
TensorBoard
Safetensors
roberta
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
dtorber commited on
Commit
4fe8e51
•
1 Parent(s): 7dae00d

Model save

Browse files
Files changed (2) hide show
  1. README.md +16 -17
  2. model.safetensors +1 -1
README.md CHANGED
@@ -17,14 +17,14 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [vinai/bertweet-base](https://huggingface.co/vinai/bertweet-base) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.5822
21
- - F1 Macro: 0.7750
22
- - F1: 0.8571
23
- - F1 Neg: 0.6929
24
- - Acc: 0.805
25
- - Prec: 0.8069
26
- - Recall: 0.9141
27
- - Mcc: 0.5646
28
 
29
  ## Model description
30
 
@@ -47,7 +47,6 @@ The following hyperparameters were used during training:
47
  - train_batch_size: 8
48
  - eval_batch_size: 8
49
  - seed: 42
50
- - distributed_type: multi-GPU
51
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
52
  - lr_scheduler_type: linear
53
  - num_epochs: 5
@@ -57,16 +56,16 @@ The following hyperparameters were used during training:
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | F1 Macro | F1 | F1 Neg | Acc | Prec | Recall | Mcc |
59
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:------:|:------:|:------:|:------:|:------:|
60
- | 0.6482 | 1.0 | 600 | 0.5610 | 0.6822 | 0.8022 | 0.5622 | 0.7275 | 0.7492 | 0.8633 | 0.3812 |
61
- | 0.5567 | 2.0 | 1200 | 0.5125 | 0.7382 | 0.8364 | 0.64 | 0.775 | 0.7823 | 0.8984 | 0.4938 |
62
- | 0.4549 | 3.0 | 1800 | 0.6164 | 0.7195 | 0.7588 | 0.6802 | 0.725 | 0.865 | 0.6758 | 0.4688 |
63
- | 0.4021 | 4.0 | 2400 | 0.5785 | 0.7344 | 0.7875 | 0.6812 | 0.745 | 0.8438 | 0.7383 | 0.4789 |
64
- | 0.3159 | 5.0 | 3000 | 0.5822 | 0.7750 | 0.8571 | 0.6929 | 0.805 | 0.8069 | 0.9141 | 0.5646 |
65
 
66
 
67
  ### Framework versions
68
 
69
- - Transformers 4.38.2
70
- - Pytorch 2.2.1+cu121
71
  - Datasets 2.18.0
72
- - Tokenizers 0.15.2
 
17
 
18
  This model is a fine-tuned version of [vinai/bertweet-base](https://huggingface.co/vinai/bertweet-base) on the None dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.6796
21
+ - F1 Macro: 0.8476
22
+ - F1: 0.8811
23
+ - F1 Neg: 0.8141
24
+ - Acc: 0.855
25
+ - Prec: 0.9267
26
+ - Recall: 0.8398
27
+ - Mcc: 0.7020
28
 
29
  ## Model description
30
 
 
47
  - train_batch_size: 8
48
  - eval_batch_size: 8
49
  - seed: 42
 
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: linear
52
  - num_epochs: 5
 
56
 
57
  | Training Loss | Epoch | Step | Validation Loss | F1 Macro | F1 | F1 Neg | Acc | Prec | Recall | Mcc |
58
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:------:|:------:|:------:|:------:|:------:|
59
+ | 0.6221 | 1.0 | 1161 | 0.5233 | 0.7216 | 0.8315 | 0.6116 | 0.765 | 0.7682 | 0.9062 | 0.4689 |
60
+ | 0.4332 | 2.0 | 2322 | 0.4843 | 0.7862 | 0.8680 | 0.7045 | 0.8175 | 0.8081 | 0.9375 | 0.5946 |
61
+ | 0.3714 | 3.0 | 3483 | 0.5872 | 0.8405 | 0.8963 | 0.7846 | 0.86 | 0.8521 | 0.9453 | 0.6914 |
62
+ | 0.2856 | 4.0 | 4644 | 0.5511 | 0.8589 | 0.8984 | 0.8194 | 0.87 | 0.8984 | 0.8984 | 0.7179 |
63
+ | 0.2199 | 5.0 | 5805 | 0.6796 | 0.8476 | 0.8811 | 0.8141 | 0.855 | 0.9267 | 0.8398 | 0.7020 |
64
 
65
 
66
  ### Framework versions
67
 
68
+ - Transformers 4.40.1
69
+ - Pytorch 2.3.0+cu121
70
  - Datasets 2.18.0
71
+ - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e8346302fb92cbbef57abfb6de3e86591c9bb3d2a4d3bec68c662a23435ad0d8
3
  size 539629568
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e99a96d594ad67f9adaa10ba707d8ab6f6be843f2055ec56d301a09d701787b
3
  size 539629568