update model card README.md

README.md

@@ -5,9 +5,24 @@ tags:
 - generated_from_trainer
 datasets:
 - imagefolder
+metrics:
+- accuracy
 model-index:
 - name: attraction-classifier
-  results: []
+  results:
+  - task:
+      name: Image Classification
+      type: image-classification
+    dataset:
+      name: imagefolder
+      type: imagefolder
+      config: smtn_girls_likeOrNot
+      split: train
+      args: smtn_girls_likeOrNot
+    metrics:
+    - name: Accuracy
+      type: accuracy
+      value: 0.8284457478005866
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,6 +31,9 @@ should probably proofread and complete it, then remove this comment. -->
 # attraction-classifier
 
 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: 0.4361
+- Accuracy: 0.8284
 
 ## Model description
 
@@ -45,6 +63,22 @@ The following hyperparameters were used during training:
 - lr_scheduler_warmup_ratio: 0.1
 - num_epochs: 10
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.6014        | 0.98  | 42   | 0.5286          | 0.7507   |
+| 0.4479        | 1.99  | 85   | 0.4547          | 0.8094   |
+| 0.3988        | 2.99  | 128  | 0.4259          | 0.8284   |
+| 0.3773        | 4.0   | 171  | 0.4475          | 0.7962   |
+| 0.3217        | 4.98  | 213  | 0.4155          | 0.8226   |
+| 0.2844        | 5.99  | 256  | 0.4423          | 0.8065   |
+| 0.2519        | 6.99  | 299  | 0.4961          | 0.8065   |
+| 0.2527        | 8.0   | 342  | 0.4642          | 0.8123   |
+| 0.2165        | 8.98  | 384  | 0.4860          | 0.8050   |
+| 0.2323        | 9.82  | 420  | 0.4361          | 0.8284   |
+
+
 ### Framework versions
 
 - Transformers 4.31.0