End of training

README.md

@@ -4,9 +4,26 @@ license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
 - generated_from_trainer
+datasets:
+- imagefolder
+metrics:
+- accuracy
 model-index:
 - name: lettuce-npk-vit
-  results: []
+  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.7619047619047619
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,7 +32,10 @@ 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/abdoulaye-diop/lettuce-npk-deficiency-prediction/runs/at88jlqw)
 # lettuce-npk-vit
 
-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.
+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.
+It achieves the following results on the evaluation set:
+- Loss: 1.0807
+- Accuracy: 0.7619
 
 ## Model description
 
@@ -45,6 +65,16 @@ The following hyperparameters were used during training:
 - lr_scheduler_warmup_ratio: 0.1
 - num_epochs: 5
 
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss | Accuracy |
+|:-------------:|:------:|:----:|:---------------:|:--------:|
+| No log        | 0.7273 | 2    | 1.3188          | 0.5238   |
+| No log        | 1.8182 | 5    | 1.1784          | 0.7381   |
+| No log        | 2.9091 | 8    | 1.1141          | 0.7857   |
+| 1.164         | 3.6364 | 10   | 1.0807          | 0.7619   |
+
+
 ### Framework versions
 
 - Transformers 4.44.2