Hugging Face's logo

mateiaassAI
/
teacher_sst2_redv2

Text Classification
Transformers
Safetensors
bert
Generated from Trainer
text-embeddings-inference
Model card Files
xet
 Community
mateiaassAI commited on
Commit
593cbee
·
verified ·
1 Parent(s): 14642c8

End of training

Browse files
Files changed (3) hide show
  1. README.md +12 -15
  2. model.safetensors +1 -1
  3. training_args.bin +1 -1
README.md CHANGED
@@ -21,12 +21,12 @@ should probably proofread and complete it, then remove this comment. -->
21
 
22
  This model is a fine-tuned version of [mateiaassAI/teacher_sst2](https://huggingface.co/mateiaassAI/teacher_sst2) on an unknown dataset.
23
  It achieves the following results on the evaluation set:
24
- - Loss: 0.2859
25
- - F1: 0.6760
26
- - Roc Auc: 0.8019
27
- - Accuracy: 0.5654
28
- - Precision: 0.6859
29
- - Recall: 0.6688
30
 
31
  ## Model description
32
 
@@ -45,25 +45,22 @@ More information needed
45
  ### Training hyperparameters
46
 
47
  The following hyperparameters were used during training:
48
- - learning_rate: 2e-05
49
  - train_batch_size: 16
50
  - eval_batch_size: 16
51
  - seed: 42
52
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
53
  - lr_scheduler_type: linear
54
- - num_epochs: 7
55
 
56
  ### Training results
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | F1 | Roc Auc | Accuracy | Precision | Recall |
59
  |:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|:--------:|:---------:|:------:|
60
- | No log | 1.0 | 256 | 0.2532 | 0.6291 | 0.7518 | 0.5120 | 0.8072 | 0.5367 |
61
- | 0.2808 | 2.0 | 512 | 0.2475 | 0.6586 | 0.7852 | 0.5599 | 0.7466 | 0.6150 |
62
- | 0.2808 | 3.0 | 768 | 0.2566 | 0.6684 | 0.7918 | 0.5875 | 0.7211 | 0.6395 |
63
- | 0.138 | 4.0 | 1024 | 0.2618 | 0.6808 | 0.7942 | 0.5838 | 0.7386 | 0.6460 |
64
- | 0.138 | 5.0 | 1280 | 0.2717 | 0.6767 | 0.7944 | 0.5727 | 0.7130 | 0.6476 |
65
- | 0.0761 | 6.0 | 1536 | 0.2858 | 0.6671 | 0.7956 | 0.5562 | 0.6882 | 0.6525 |
66
- | 0.0761 | 7.0 | 1792 | 0.2859 | 0.6760 | 0.8019 | 0.5654 | 0.6859 | 0.6688 |
67
 
68
 
69
  ### Framework versions
 
21
 
22
  This model is a fine-tuned version of [mateiaassAI/teacher_sst2](https://huggingface.co/mateiaassAI/teacher_sst2) on an unknown dataset.
23
  It achieves the following results on the evaluation set:
24
+ - Loss: 0.2390
25
+ - F1: 0.6897
26
+ - Roc Auc: 0.7984
27
+ - Accuracy: 0.5893
28
+ - Precision: 0.7488
29
+ - Recall: 0.6476
30
 
31
  ## Model description
32
 
 
45
  ### Training hyperparameters
46
 
47
  The following hyperparameters were used during training:
48
+ - learning_rate: 1.7e-05
49
  - train_batch_size: 16
50
  - eval_batch_size: 16
51
  - seed: 42
52
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
53
  - lr_scheduler_type: linear
54
+ - num_epochs: 4
55
 
56
  ### Training results
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | F1 | Roc Auc | Accuracy | Precision | Recall |
59
  |:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|:--------:|:---------:|:------:|
60
+ | No log | 1.0 | 256 | 0.2659 | 0.5655 | 0.7202 | 0.4530 | 0.8249 | 0.4698 |
61
+ | 0.2931 | 2.0 | 512 | 0.2460 | 0.6641 | 0.7856 | 0.5635 | 0.7677 | 0.6101 |
62
+ | 0.2931 | 3.0 | 768 | 0.2398 | 0.6791 | 0.7926 | 0.5764 | 0.7468 | 0.6330 |
63
+ | 0.1701 | 4.0 | 1024 | 0.2390 | 0.6897 | 0.7984 | 0.5893 | 0.7488 | 0.6476 |
 
 
 
64
 
65
 
66
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b9c1211bdade2948d9d4d97a541faa40cd5da9fc9e2a55435846880fc060126f
3
  size 497810452
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:079eb4959804fcd079603651f353558b7de2d046df077ef2a02de41afc242729
3
  size 497810452
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bae74e979f9a735708febf36b002d268d5368360c19e1ea85a2306754d8fb25e
3
  size 5176
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80da1db41164565dd8e55faee1bcf069e9ba0d5a6dc9a5112646ac798d0a769e
3
  size 5176