End of training

README.md

@@ -3,26 +3,11 @@ license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
 - generated_from_trainer
-datasets:
-- imagefolder
 metrics:
 - accuracy
 model-index:
 - name: chest_xray_pneumonia
-  results:
-  - task:
-      name: Image Classification
-      type: image-classification
-    dataset:
-      name: imagefolder
-      type: imagefolder
-      config: default
-      split: test
-      args: default
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.8894230769230769
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,10 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # chest_xray_pneumonia
 
-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 imagefolder dataset.
+This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3211
-- Accuracy: 0.8894
+- Loss: 0.3081
+- Accuracy: 0.8942
 
 ## Model description
 
@@ -67,14 +52,14 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.1715        | 0.99  | 81   | 0.4023          | 0.8590   |
-| 0.129         | 2.0   | 163  | 0.2389          | 0.9263   |
-| 0.1064        | 2.98  | 243  | 0.3211          | 0.8894   |
+| 0.1991        | 0.99  | 81   | 0.4470          | 0.8510   |
+| 0.1574        | 2.0   | 163  | 0.3711          | 0.8926   |
+| 0.1068        | 2.98  | 243  | 0.3081          | 0.8942   |
 
 
 ### Framework versions
 
-- Transformers 4.36.1
-- Pytorch 2.1.2+cu121
-- Datasets 2.15.0
+- Transformers 4.35.2
+- Pytorch 2.1.0+cu121
+- Datasets 2.16.0
 - Tokenizers 0.15.0