Hugging Face's logo

ongkn
/
attraction-classifier

Image Classification
Transformers
Safetensors
vit
Generated from Trainer
Eval Results
Inference Endpoints
Model card Files Community
5
ongkn commited on
Commit
7f973c3
·
1 Parent(s): 48f85ab

End of training

Browse files
Files changed (2) hide show
  1. README.md +37 -36
  2. model.safetensors +1 -1
README.md CHANGED
@@ -5,9 +5,24 @@ tags:
5
  - generated_from_trainer
6
  datasets:
7
  - imagefolder
 
 
8
  model-index:
9
  - name: attraction-classifier
10
- results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  ---
12
 
13
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -17,7 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  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.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 0.5429
 
21
 
22
  ## Model description
23
 
@@ -47,40 +63,25 @@ The following hyperparameters were used during training:
47
 
48
  ### Training results
49
 
50
- | Training Loss | Epoch | Step | Validation Loss |
51
- |:-------------:|:-----:|:----:|:---------------:|
52
- | 0.7599 | 0.11 | 20 | 0.7105 |
53
- | 0.6535 | 0.22 | 40 | 0.6561 |
54
- | 0.6191 | 0.33 | 60 | 0.6045 |
55
- | 0.7219 | 0.44 | 80 | 0.5976 |
56
- | 0.62 | 0.55 | 100 | 0.5781 |
57
- | 0.5249 | 0.66 | 120 | 0.5743 |
58
- | 0.5934 | 0.77 | 140 | 0.5684 |
59
- | 0.4143 | 0.88 | 160 | 0.5522 |
60
- | 0.4472 | 0.99 | 180 | 0.5645 |
61
- | 0.3507 | 1.1 | 200 | 0.5784 |
62
- | 0.4165 | 1.21 | 220 | 0.5796 |
63
- | 0.2822 | 1.32 | 240 | 0.5867 |
64
- | 0.4765 | 1.43 | 260 | 0.5446 |
65
- | 0.4557 | 1.54 | 280 | 0.5739 |
66
- | 0.4087 | 1.65 | 300 | 0.5292 |
67
- | 0.4181 | 1.76 | 320 | 0.5182 |
68
- | 0.6616 | 1.87 | 340 | 0.5328 |
69
- | 0.48 | 1.98 | 360 | 0.5282 |
70
- | 0.2159 | 2.09 | 380 | 0.5457 |
71
- | 0.5038 | 2.2 | 400 | 0.5832 |
72
- | 0.6434 | 2.31 | 420 | 0.5667 |
73
- | 0.4566 | 2.42 | 440 | 0.5128 |
74
- | 0.2972 | 2.53 | 460 | 0.5475 |
75
- | 0.5094 | 2.64 | 480 | 0.5536 |
76
- | 0.4807 | 2.75 | 500 | 0.5512 |
77
- | 0.3666 | 2.86 | 520 | 0.5128 |
78
- | 0.38 | 2.97 | 540 | 0.5499 |
79
- | 0.3496 | 3.08 | 560 | 0.5129 |
80
- | 0.3503 | 3.19 | 580 | 0.5480 |
81
- | 0.4435 | 3.3 | 600 | 0.5918 |
82
- | 0.2163 | 3.41 | 620 | 0.5558 |
83
- | 0.4316 | 3.52 | 640 | 0.5429 |
84
 
85
 
86
  ### Framework versions
 
5
  - generated_from_trainer
6
  datasets:
7
  - imagefolder
8
+ metrics:
9
+ - accuracy
10
  model-index:
11
  - name: attraction-classifier
12
+ results:
13
+ - task:
14
+ name: Image Classification
15
+ type: image-classification
16
+ dataset:
17
+ name: imagefolder
18
+ type: imagefolder
19
+ config: default
20
+ split: train
21
+ args: default
22
+ metrics:
23
+ - name: Accuracy
24
+ type: accuracy
25
+ value: 0.7554179566563467
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
32
 
33
  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.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.6591
36
+ - Accuracy: 0.7554
37
 
38
  ## Model description
39
 
 
63
 
64
  ### Training results
65
 
66
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
+ | 0.6302 | 0.55 | 100 | 0.5796 | 0.6997 |
69
+ | 0.6054 | 1.1 | 200 | 0.5893 | 0.7307 |
70
+ | 0.4672 | 1.65 | 300 | 0.5445 | 0.7461 |
71
+ | 0.5753 | 2.2 | 400 | 0.5656 | 0.7214 |
72
+ | 0.3311 | 2.75 | 500 | 0.5671 | 0.6811 |
73
+ | 0.2875 | 3.3 | 600 | 0.5796 | 0.6873 |
74
+ | 0.436 | 3.85 | 700 | 0.5204 | 0.7492 |
75
+ | 0.3871 | 4.4 | 800 | 0.5480 | 0.7276 |
76
+ | 0.3486 | 4.95 | 900 | 0.5730 | 0.7523 |
77
+ | 0.2168 | 5.49 | 1000 | 0.5245 | 0.7399 |
78
+ | 0.2935 | 6.04 | 1100 | 0.5652 | 0.7678 |
79
+ | 0.5159 | 6.59 | 1200 | 0.5320 | 0.7709 |
80
+ | 0.3764 | 7.14 | 1300 | 0.5471 | 0.7709 |
81
+ | 0.1869 | 7.69 | 1400 | 0.6399 | 0.7430 |
82
+ | 0.2821 | 8.24 | 1500 | 0.7145 | 0.7492 |
83
+ | 0.1765 | 8.79 | 1600 | 0.6636 | 0.7802 |
84
+ | 0.0367 | 9.34 | 1700 | 0.6591 | 0.7554 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
85
 
86
 
87
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:09215d123f7fb038479b8fdc2db57c0a79fb7a136c300ef4667587809880c7ad
3
  size 343223968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3b9596c2eb8b0371b79b90adb4ae520bf62870173ba1d612494fe04efeb901e
3
  size 343223968