Hugging Face's logo

synergyai-jaeung
/
vit-base-patch16-224-in21k_covid_19_ct_scans

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Eval Results
Inference Endpoints
Model card Files Metrics Community
synergyai-jaeung commited on
Commit
e141828
•
1 Parent(s): 77ee1cd

Model save

Browse files
Files changed (3) hide show
  1. README.md +26 -23
  2. model.safetensors +1 -1
  3. runs/May28_16-57-52_RTX3090/events.out.tfevents.1716883073.RTX3090.462376.0 +2 -2
README.md CHANGED
@@ -25,16 +25,16 @@ model-index:
25
  metrics:
26
  - name: Accuracy
27
  type: accuracy
28
- value: 0.9197530864197531
29
  - name: F1
30
  type: f1
31
- value: 0.6365832614322692
32
  - name: Recall
33
  type: recall
34
- value: 0.9931972789115646
35
  - name: Precision
36
  type: precision
37
- value: 0.9240506329113924
38
  ---
39
 
40
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -44,12 +44,12 @@ should probably proofread and complete it, then remove this comment. -->
44
 
45
  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.
46
  It achieves the following results on the evaluation set:
47
- - Loss: 0.5002
48
- - Accuracy: 0.9198
49
- - F1: 0.6366
50
- - Auc: 0.5966
51
- - Recall: 0.9932
52
- - Precision: 0.9241
53
 
54
  ## Model description
55
 
@@ -74,24 +74,27 @@ The following hyperparameters were used during training:
74
  - seed: 42
75
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
76
  - lr_scheduler_type: linear
77
- - num_epochs: 12
78
 
79
  ### Training results
80
 
81
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Auc | Recall | Precision |
82
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:------:|:------:|:---------:|
83
- | 0.7767 | 1.0 | 47 | 0.3346 | 0.9074 | 0.4757 | 0.5 | 1.0 | 0.9074 |
84
- | 0.7767 | 2.0 | 94 | 0.5513 | 0.8272 | 0.5919 | 0.6204 | 0.8741 | 0.9312 |
85
- | 0.7767 | 3.0 | 141 | 0.4290 | 0.9074 | 0.4757 | 0.5 | 1.0 | 0.9074 |
86
- | 0.7767 | 4.0 | 188 | 0.4333 | 0.9198 | 0.6366 | 0.5966 | 0.9932 | 0.9241 |
87
- | 0.7767 | 5.0 | 235 | 0.5041 | 0.9074 | 0.6181 | 0.5898 | 0.9796 | 0.9231 |
88
- | 0.7767 | 6.0 | 282 | 0.4848 | 0.9167 | 0.6317 | 0.5949 | 0.9898 | 0.9238 |
89
- | 0.7767 | 7.0 | 329 | 0.4877 | 0.9198 | 0.6366 | 0.5966 | 0.9932 | 0.9241 |
90
- | 0.7767 | 8.0 | 376 | 0.4926 | 0.9198 | 0.6366 | 0.5966 | 0.9932 | 0.9241 |
91
- | 0.7767 | 9.0 | 423 | 0.4958 | 0.9198 | 0.6366 | 0.5966 | 0.9932 | 0.9241 |
92
- | 0.7767 | 10.0 | 470 | 0.4981 | 0.9198 | 0.6366 | 0.5966 | 0.9932 | 0.9241 |
93
- | 0.0381 | 11.0 | 517 | 0.4996 | 0.9198 | 0.6366 | 0.5966 | 0.9932 | 0.9241 |
94
- | 0.0381 | 12.0 | 564 | 0.5002 | 0.9198 | 0.6366 | 0.5966 | 0.9932 | 0.9241 |
 
 
 
95
 
96
 
97
  ### Framework versions
 
25
  metrics:
26
  - name: Accuracy
27
  type: accuracy
28
+ value: 0.9010416666666666
29
  - name: F1
30
  type: f1
31
+ value: 0.473972602739726
32
  - name: Recall
33
  type: recall
34
+ value: 0.9942528735632183
35
  - name: Precision
36
  type: precision
37
+ value: 0.9057591623036649
38
  ---
39
 
40
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
44
 
45
  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.
46
  It achieves the following results on the evaluation set:
47
+ - Loss: 0.6385
48
+ - Accuracy: 0.9010
49
+ - F1: 0.4740
50
+ - Auc: 0.4971
51
+ - Recall: 0.9943
52
+ - Precision: 0.9058
53
 
54
  ## Model description
55
 
 
74
  - seed: 42
75
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
76
  - lr_scheduler_type: linear
77
+ - num_epochs: 15
78
 
79
  ### Training results
80
 
81
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Auc | Recall | Precision |
82
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:------:|:------:|:---------:|
83
+ | 0.7218 | 1.0 | 55 | 0.3383 | 0.9062 | 0.4754 | 0.5 | 1.0 | 0.9062 |
84
+ | 0.7218 | 2.0 | 110 | 0.3823 | 0.9062 | 0.4754 | 0.5 | 1.0 | 0.9062 |
85
+ | 0.7218 | 3.0 | 165 | 0.3957 | 0.9062 | 0.4754 | 0.5 | 1.0 | 0.9062 |
86
+ | 0.7218 | 4.0 | 220 | 0.4485 | 0.9062 | 0.4754 | 0.5 | 1.0 | 0.9062 |
87
+ | 0.7218 | 5.0 | 275 | 0.4786 | 0.8958 | 0.4725 | 0.4943 | 0.9885 | 0.9053 |
88
+ | 0.7218 | 6.0 | 330 | 0.5316 | 0.9010 | 0.4740 | 0.4971 | 0.9943 | 0.9058 |
89
+ | 0.7218 | 7.0 | 385 | 0.5539 | 0.9010 | 0.4740 | 0.4971 | 0.9943 | 0.9058 |
90
+ | 0.7218 | 8.0 | 440 | 0.5800 | 0.9010 | 0.4740 | 0.4971 | 0.9943 | 0.9058 |
91
+ | 0.7218 | 9.0 | 495 | 0.5977 | 0.9010 | 0.4740 | 0.4971 | 0.9943 | 0.9058 |
92
+ | 0.0987 | 10.0 | 550 | 0.6110 | 0.9010 | 0.4740 | 0.4971 | 0.9943 | 0.9058 |
93
+ | 0.0987 | 11.0 | 605 | 0.6211 | 0.9010 | 0.4740 | 0.4971 | 0.9943 | 0.9058 |
94
+ | 0.0987 | 12.0 | 660 | 0.6288 | 0.9010 | 0.4740 | 0.4971 | 0.9943 | 0.9058 |
95
+ | 0.0987 | 13.0 | 715 | 0.6341 | 0.9010 | 0.4740 | 0.4971 | 0.9943 | 0.9058 |
96
+ | 0.0987 | 14.0 | 770 | 0.6374 | 0.9010 | 0.4740 | 0.4971 | 0.9943 | 0.9058 |
97
+ | 0.0987 | 15.0 | 825 | 0.6385 | 0.9010 | 0.4740 | 0.4971 | 0.9943 | 0.9058 |
98
 
99
 
100
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:93e89efe9d39ff1dc5b701b1fa8faf267eb16cba584a157772a54f39a06356b0
3
  size 343223968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:760373362287b69643cb264294c3cdca3c69105ed8734ed2b2cb181e5b3d2411
3
  size 343223968
runs/May28_16-57-52_RTX3090/events.out.tfevents.1716883073.RTX3090.462376.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0a5d72fb449a2590d562ea8e86375d4aac65d32421182b75a47723608e37ad13
3
- size 12582
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5bff40d0d524fcd00ce0adff172c20e201a4fafc73f774f6951099f3ab77392c
3
+ size 13455