selmamalak
/

organc-vit-base-finetuned

PEFT

Safetensors

Generated from Trainer

Model card Files Files and versions Community

selmamalak commited on Apr 13, 2024

Commit

e11e64a

verified ·

1 Parent(s): 35616b1

Model save

Browse files

Files changed (1) hide show

README.md +17 -17

README.md CHANGED Viewed

@@ -12,22 +12,22 @@ metrics:
 - f1
 base_model: google/vit-base-patch16-224-in21k
 model-index:
-- name: organc-beit-base-finetuned
   results: []
 ---
 <!-- 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. -->
-# organc-beit-base-finetuned
 This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the medmnist-v2 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2607
-- Accuracy: 0.9128
-- Precision: 0.9094
-- Recall: 0.8976
-- F1: 0.9019
 ## Model description
@@ -61,16 +61,16 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
-| 0.7007        | 1.0   | 203  | 0.2457          | 0.9114   | 0.9019    | 0.8644 | 0.8529 |
-| 0.6322        | 2.0   | 406  | 0.2148          | 0.9423   | 0.9424    | 0.9292 | 0.9340 |
-| 0.6353        | 3.0   | 609  | 0.1218          | 0.9632   | 0.9546    | 0.9542 | 0.9529 |
-| 0.6176        | 4.0   | 813  | 0.0839          | 0.9799   | 0.9775    | 0.9793 | 0.9782 |
-| 0.4913        | 5.0   | 1016 | 0.1008          | 0.9712   | 0.9713    | 0.9717 | 0.9707 |
-| 0.4943        | 6.0   | 1219 | 0.0805          | 0.9799   | 0.9843    | 0.9820 | 0.9828 |
-| 0.426         | 7.0   | 1422 | 0.0671          | 0.9799   | 0.9782    | 0.9787 | 0.9781 |
-| 0.4152        | 8.0   | 1626 | 0.0566          | 0.9870   | 0.9862    | 0.9896 | 0.9878 |
-| 0.2927        | 9.0   | 1829 | 0.0658          | 0.9837   | 0.9850    | 0.9850 | 0.9848 |
-| 0.3154        | 9.99  | 2030 | 0.0537          | 0.9841   | 0.9855    | 0.9854 | 0.9853 |
 ### Framework versions

 - f1
 base_model: google/vit-base-patch16-224-in21k
 model-index:
+- name: organc-vit-base-finetuned
   results: []
 ---
 <!-- 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. -->
+# organc-vit-base-finetuned
 This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the medmnist-v2 dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.0732
+- Accuracy: 0.9808
+- Precision: 0.9830
+- Recall: 0.9826
+- F1: 0.9825
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
+| 0.6525        | 1.0   | 203  | 0.2025          | 0.9327   | 0.9260    | 0.9130 | 0.9091 |
+| 0.765         | 2.0   | 406  | 0.2110          | 0.9377   | 0.9441    | 0.9289 | 0.9344 |
+| 0.6514        | 3.0   | 609  | 0.2026          | 0.9490   | 0.9457    | 0.9442 | 0.9428 |
+| 0.6405        | 4.0   | 813  | 0.2056          | 0.9289   | 0.9481    | 0.9175 | 0.9267 |
+| 0.6514        | 5.0   | 1016 | 0.1362          | 0.9523   | 0.9459    | 0.9385 | 0.9382 |
+| 0.5778        | 6.0   | 1219 | 0.0787          | 0.9770   | 0.9739    | 0.9746 | 0.9737 |
+| 0.4759        | 7.0   | 1422 | 0.0959          | 0.9724   | 0.9744    | 0.9693 | 0.9714 |
+| 0.482         | 8.0   | 1626 | 0.0743          | 0.9762   | 0.9737    | 0.9737 | 0.9733 |
+| 0.3729        | 9.0   | 1829 | 0.0903          | 0.9758   | 0.9778    | 0.9754 | 0.9762 |
+| 0.3705        | 9.99  | 2030 | 0.0732          | 0.9808   | 0.9830    | 0.9826 | 0.9825 |
 ### Framework versions