Model save

README.md

@@ -3,26 +3,9 @@ license: apache-2.0
 base_model: facebook/deit-base-distilled-patch16-224
 tags:
 - generated_from_trainer
-datasets:
-- imagefolder
-metrics:
-- accuracy
 model-index:
 - name: alzheimer_model_aug_deit5
-  results:
-  - task:
-      name: Image Classification
-      type: image-classification
-    dataset:
-      name: imagefolder
-      type: imagefolder
-      config: default
-      split: train
-      args: default
-    metrics:
-    - name: Accuracy
-      type: accuracy
-      value: 0.9848812095032398
+  results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,10 +13,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # alzheimer_model_aug_deit5
 
-This model is a fine-tuned version of [facebook/deit-base-distilled-patch16-224](https://huggingface.co/facebook/deit-base-distilled-patch16-224) on the imagefolder dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.0557
-- Accuracy: 0.9849
+This model is a fine-tuned version of [facebook/deit-base-distilled-patch16-224](https://huggingface.co/facebook/deit-base-distilled-patch16-224) on an unknown dataset.
 
 ## Model description
 
@@ -62,17 +42,6 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 5
 
-### Training results
-
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 0.95  | 11   | 0.6706          | 0.8963   |
-| No log        | 1.98  | 23   | 0.1733          | 0.9633   |
-| No log        | 2.93  | 34   | 0.0929          | 0.9762   |
-| No log        | 3.97  | 46   | 0.0724          | 0.9762   |
-| No log        | 4.74  | 55   | 0.0557          | 0.9849   |
-
-
 ### Framework versions
 
 - Transformers 4.35.2