End of training

README.md

@@ -17,16 +17,17 @@ 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)
 # fold_0
 
 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: 1.5239
-- Precision: 0.3745
-- Recall: 0.9507
-- F1: 0.7320
-- Pr Auc: 0.0641
-- Roc Auc: 0.6223
+- Loss: 0.0130
+- Precision: 0.1834
+- Recall: 0.3042
+- F1: 0.5248
+- Pr Auc: 0.6999
+- Roc Auc: 0.9149
 
 ## Model description
 
@@ -51,10 +52,17 @@ 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: 0
+- num_epochs: 5
 
 ### Training results
 
+| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Pr Auc | Roc Auc |
+|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:------:|:-------:|
+| 0.0281        | 1.0   | 630  | 0.0113          | 0.3745    | 0.9507 | 0.7320 | 0.6813 | 0.9278  |
+| 0.009         | 2.0   | 1260 | 0.0108          | 0.5987    | 0.1560 | 0.1560 | 0.5519 | 0.9242  |
+| 0.0061        | 3.0   | 1890 | 0.0110          | 0.0581    | 0.8662 | 0.6011 | 0.6682 | 0.9283  |
+| 0.0012        | 4.0   | 2520 | 0.0122          | 0.7081    | 0.0206 | 0.9699 | 0.7090 | 0.9146  |
+| 0.0006        | 5.0   | 3150 | 0.0130          | 0.8324    | 0.2123 | 0.1818 | 0.6999 | 0.9149  |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0c52ca463b185c829f5002d48097e198b2f756bae22338c9376c5c3c309b01dc
+oid sha256:23674e7d194dfab016dad3cbc07913cb6b7af01cec70bfd5a6137eaee32105eb
 size 554446244

runs/May12_06-40-35_aff77f7d1741/events.out.tfevents.1715498849.aff77f7d1741.6664.6

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:816a66e6602dbe7d2c8b020c3bfde8fa3dde6a97ec74484e5e5ab8144b799935
+size 609

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:549926a12155753ea86f7842773f5592f60b073147b1af81e723a84545ccb770
+oid sha256:54ef968038b58946a8696ca92ec304ff614c64273f63d01caca5045f8b1d9e44
 size 5176