End of training

README.md

@@ -7,6 +7,7 @@ metrics:
 - precision
 - recall
 - f1
+- accuracy
 model-index:
 - name: fold_3
   results: []
@@ -15,22 +16,24 @@ 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/amnasaeed100/FineTuning-ADE-change1/runs/qyo3k3m3)
+[<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)
+[<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/10xzvwgi)
 # fold_3
 
 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.0185
-- Precision: 0.1834
-- Recall: 0.3042
-- F1: 0.5248
-- Pr Auc: 0.6009
-- Roc Auc: 0.8987
+- Loss: 0.0125
+- Precision: 0.6887
+- Recall: 0.6744
+- F1: 0.6814
+- Accuracy: 0.9992
+- Roc Auc: 0.9939
+- Pr Auc: 0.9998
 
 ## Model description
 
@@ -55,17 +58,20 @@ 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.0306        | 1.0   | 630  | 0.0134          | 0.3745    | 0.9507 | 0.7320 | 0.6102 | 0.9543  |
-| 0.01          | 2.0   | 1260 | 0.0119          | 0.5987    | 0.1560 | 0.1560 | 0.5802 | 0.9372  |
-| 0.0057        | 3.0   | 1890 | 0.0131          | 0.0581    | 0.8662 | 0.6011 | 0.5880 | 0.9473  |
-| 0.0013        | 4.0   | 2520 | 0.0173          | 0.7081    | 0.0206 | 0.9699 | 0.6219 | 0.9066  |
-| 0.0006        | 5.0   | 3150 | 0.0185          | 0.8324    | 0.2123 | 0.1818 | 0.6009 | 0.8987  |
+| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy | Roc Auc | Pr Auc |
+|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|:-------:|:------:|
+| 0.0286        | 1.0   | 632  | 0.0159          | 0.5908    | 0.5035 | 0.5436 | 0.9988   | 0.9931  | 0.9998 |
+| 0.0115        | 2.0   | 1264 | 0.0125          | 0.6887    | 0.6744 | 0.6814 | 0.9992   | 0.9939  | 0.9998 |
+| 0.0079        | 3.0   | 1896 | 0.0170          | 0.8419    | 0.5289 | 0.6496 | 0.9992   | 0.9859  | 0.9995 |
+| 0.0035        | 4.0   | 2528 | 0.0150          | 0.7146    | 0.7344 | 0.7244 | 0.9992   | 0.9903  | 0.9997 |
+| 0.002         | 5.0   | 3160 | 0.0166          | 0.6471    | 0.7621 | 0.6999 | 0.9991   | 0.9917  | 0.9997 |
+| 0.0017        | 6.0   | 3792 | 0.0196          | 0.8300    | 0.6651 | 0.7385 | 0.9993   | 0.9865  | 0.9995 |
+| 0.0012        | 7.0   | 4424 | 0.0175          | 0.7143    | 0.7621 | 0.7374 | 0.9993   | 0.9920  | 0.9997 |
+| 0.0004        | 8.0   | 5056 | 0.0176          | 0.7262    | 0.7413 | 0.7337 | 0.9992   | 0.9920  | 0.9997 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:514365a320952bfab0109eb41a4142f5ce46527bbcb483027da10000a93fce1a
+oid sha256:f5c3d2a426d3a50819eedfdf2ce22e8de8770ec8c2d6df293bfa85e0e8e33891
 size 554446244