Amna100 commited on
Commit
c4cd83e
1 Parent(s): 6d1b75a

End of training

Browse files
Files changed (2) hide show
  1. README.md +18 -16
  2. model.safetensors +1 -1
README.md CHANGED
@@ -16,19 +16,21 @@ model-index:
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
- [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/amnasaeed100/FineTuning-ADE-Repeatedfold/runs/lvieenf2)
20
- [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/amnasaeed100/FineTuning-ADE-Repeatedfold/runs/fgis28rc)
 
 
21
  # fold_1
22
 
23
  This model is a fine-tuned version of [Amna100/PreTraining-MLM](https://huggingface.co/Amna100/PreTraining-MLM) on the None dataset.
24
  It achieves the following results on the evaluation set:
25
- - Loss: 0.0113
26
- - Precision: 0.8172
27
- - Recall: 0.5823
28
- - F1: 0.6801
29
  - Accuracy: 0.9993
30
- - Roc Auc: 0.9933
31
- - Pr Auc: 0.9998
32
 
33
  ## Model description
34
 
@@ -53,22 +55,22 @@ The following hyperparameters were used during training:
53
  - seed: 42
54
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
55
  - lr_scheduler_type: linear
56
- - num_epochs: 10
57
 
58
  ### Training results
59
 
60
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy | Roc Auc | Pr Auc |
61
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|:-------:|:------:|
62
- | 0.0326 | 1.0 | 632 | 0.0138 | 0.6507 | 0.4668 | 0.5436 | 0.9991 | 0.9909 | 0.9998 |
63
- | 0.0116 | 2.0 | 1264 | 0.0113 | 0.8172 | 0.5823 | 0.6801 | 0.9993 | 0.9933 | 0.9998 |
64
- | 0.0071 | 3.0 | 1896 | 0.0131 | 0.7896 | 0.5995 | 0.6816 | 0.9993 | 0.9915 | 0.9998 |
65
- | 0.0023 | 4.0 | 2528 | 0.0165 | 0.6275 | 0.6953 | 0.6597 | 0.9991 | 0.9899 | 0.9998 |
66
- | 0.0013 | 5.0 | 3160 | 0.0173 | 0.7465 | 0.6585 | 0.6997 | 0.9993 | 0.9827 | 0.9996 |
67
 
68
 
69
  ### Framework versions
70
 
71
- - Transformers 4.41.0.dev0
72
- - Pytorch 2.2.1+cu121
73
  - Datasets 2.19.1
74
  - Tokenizers 0.19.1
 
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
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/amnasaeed100/FineTuning-ADE-change2/runs/zkyqf4w8)
20
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/amnasaeed100/FineTuning-ADE-change2/runs/n6lnsbeg)
21
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/amnasaeed100/FineTuning-ADE-change2/runs/k9jhon43)
22
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/amnasaeed100/FineTuning-ADE-change2/runs/67sviuwh)
23
  # fold_1
24
 
25
  This model is a fine-tuned version of [Amna100/PreTraining-MLM](https://huggingface.co/Amna100/PreTraining-MLM) on the None dataset.
26
  It achieves the following results on the evaluation set:
27
+ - Loss: 0.0104
28
+ - Precision: 0.6792
29
+ - Recall: 0.5870
30
+ - F1: 0.6297
31
  - Accuracy: 0.9993
32
+ - Roc Auc: 0.9967
33
+ - Pr Auc: 0.9999
34
 
35
  ## Model description
36
 
 
55
  - seed: 42
56
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
57
  - lr_scheduler_type: linear
58
+ - num_epochs: 5
59
 
60
  ### Training results
61
 
62
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy | Roc Auc | Pr Auc |
63
  |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|:-------:|:------:|
64
+ | 0.0252 | 1.0 | 711 | 0.0159 | 0.4538 | 0.6413 | 0.5315 | 0.9988 | 0.9944 | 0.9998 |
65
+ | 0.0095 | 2.0 | 1422 | 0.0104 | 0.6792 | 0.5870 | 0.6297 | 0.9993 | 0.9967 | 0.9999 |
66
+ | 0.003 | 3.0 | 2133 | 0.0106 | 0.6432 | 0.6957 | 0.6684 | 0.9993 | 0.9973 | 0.9999 |
67
+ | 0.0024 | 4.0 | 2844 | 0.0126 | 0.7006 | 0.6739 | 0.6870 | 0.9994 | 0.9960 | 0.9999 |
68
+ | 0.0004 | 5.0 | 3555 | 0.0148 | 0.7239 | 0.6413 | 0.6801 | 0.9994 | 0.9954 | 0.9999 |
69
 
70
 
71
  ### Framework versions
72
 
73
+ - Transformers 4.42.0.dev0
74
+ - Pytorch 2.3.0+cu121
75
  - Datasets 2.19.1
76
  - Tokenizers 0.19.1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bbe16021d72a048b581d0296e2643a4898c40d78ec0111c6326c81d437f43620
3
  size 554446244
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9966f67f140841fb9ea53786612d3656d373b33f476a611ca7f8caa2d286f8f6
3
  size 554446244