End of training

README.md

@@ -7,6 +7,7 @@ metrics:
 - precision
 - recall
 - f1
+- accuracy
 model-index:
 - name: fold_2
   results: []
@@ -15,21 +16,23 @@ model-index:
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-[<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-change1/runs/94wgcdtp)
-[<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-change1/runs/8g0cixov)
-[<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-change1/runs/05nc4r5u)
-[<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-change1/runs/2tfkcyde)
-[<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-change1/runs/2zf1k4id)
+[<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/wbresearch/FineTuning-ADE-DropOUT/runs/rw6sjeap)
+[<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/wbresearch/FineTuning-ADE-DropOUT/runs/g4pyaj7k)
+[<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/wbresearch/FineTuning-ADE-DropOUT/runs/t4il24wd)
+[<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/wbresearch/FineTuning-ADE-DropOUT/runs/qf2ywrxq)
+[<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/wbresearch/FineTuning-ADE-DropOUT/runs/9xmjfnoc)
+[<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/wbresearch/FineTuning-ADE-DropOUT/runs/vp363qmp)
 # fold_2
 
 This model is a fine-tuned version of [Amna100/PreTraining-MLM](https://huggingface.co/Amna100/PreTraining-MLM) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0123
-- Precision: 0.1834
-- Recall: 0.3042
-- F1: 0.5248
-- Pr Auc: 0.6254
-- Roc Auc: 0.9352
+- Loss: 0.0100
+- Precision: 0.7190
+- Recall: 0.7
+- F1: 0.7094
+- Accuracy: 0.9995
+- Roc Auc: 0.9959
+- Pr Auc: 0.9999
 
 ## Model description
 
@@ -54,17 +57,21 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 5
+- num_epochs: 10
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Pr Auc | Roc Auc |
-|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:------:|:-------:|
-| 0.0324        | 1.0   | 630  | 0.0108          | 0.3745    | 0.9507 | 0.7320 | 0.5652 | 0.9573  |
-| 0.0101        | 2.0   | 1260 | 0.0101          | 0.5987    | 0.1560 | 0.1560 | 0.5987 | 0.9585  |
-| 0.0067        | 3.0   | 1890 | 0.0114          | 0.0581    | 0.8662 | 0.6011 | 0.6210 | 0.9303  |
-| 0.0015        | 4.0   | 2520 | 0.0110          | 0.7081    | 0.0206 | 0.9699 | 0.5885 | 0.9400  |
-| 0.0006        | 5.0   | 3150 | 0.0123          | 0.8324    | 0.2123 | 0.1818 | 0.6254 | 0.9352  |
+| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy | Roc Auc | Pr Auc |
+|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|:-------:|:------:|
+| 0.026         | 1.0   | 632  | 0.0136          | 0.4503    | 0.7059 | 0.5498 | 0.9989   | 0.9956  | 0.9999 |
+| 0.0109        | 2.0   | 1264 | 0.0104          | 0.7270    | 0.6029 | 0.6592 | 0.9994   | 0.9951  | 0.9999 |
+| 0.0063        | 3.0   | 1896 | 0.0100          | 0.7190    | 0.7    | 0.7094 | 0.9995   | 0.9959  | 0.9999 |
+| 0.0027        | 4.0   | 2528 | 0.0113          | 0.6872    | 0.7235 | 0.7049 | 0.9994   | 0.9942  | 0.9999 |
+| 0.0012        | 5.0   | 3160 | 0.0145          | 0.8194    | 0.5471 | 0.6561 | 0.9994   | 0.9924  | 0.9999 |
+| 0.0015        | 6.0   | 3792 | 0.0131          | 0.6952    | 0.7176 | 0.7062 | 0.9994   | 0.9935  | 0.9999 |
+| 0.0008        | 7.0   | 4424 | 0.0139          | 0.7604    | 0.7    | 0.7289 | 0.9995   | 0.9925  | 0.9999 |
+| 0.0002        | 8.0   | 5056 | 0.0163          | 0.7418    | 0.6676 | 0.7028 | 0.9994   | 0.9907  | 0.9998 |
+| 0.0001        | 9.0   | 5688 | 0.0163          | 0.6551    | 0.7206 | 0.6863 | 0.9994   | 0.9927  | 0.9999 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:013a09d9cb1c2bcb0814154c92d2df59607c1f88072aa9e593011c6abd875fb9
+oid sha256:f6fbee88daa72c14bf7603be00909d3c3f1e6c31cec3a2e2876ebf3afa7f98d5
 size 554446244