Hugging Face's logo

HorcruxNo13
/
beit-base-patch16-224

Image Classification
Transformers
PyTorch
TensorBoard
Safetensors
beit
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
HorcruxNo13 commited on
Commit
2d317e6
•
1 Parent(s): 9ec33e4

Model save

Browse files
Files changed (6) hide show
  1. README.md +44 -44
  2. config.json +4 -4
  3. model.safetensors +1 -1
  4. runs/May04_13-42-18_16ca11c6137e/events.out.tfevents.1714830139.16ca11c6137e.960.0 +2 -2
  5. runs/May04_13-42-18_16ca11c6137e/events.out.tfevents.1714831102.16ca11c6137e.960.1 +3 -0
  6. training_args.bin +1 -1
README.md CHANGED
@@ -19,11 +19,11 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  This model is a fine-tuned version of [microsoft/beit-base-patch16-224](https://huggingface.co/microsoft/beit-base-patch16-224) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.5338
23
- - Accuracy: 0.7165
24
- - Precision: 0.7127
25
- - Recall: 0.7165
26
- - F1 Score: 0.7139
27
 
28
  ## Model description
29
 
@@ -57,47 +57,47 @@ The following hyperparameters were used during training:
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 Score |
59
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:--------:|
60
- | No log | 0.8 | 2 | 0.7127 | 0.5686 | 0.3992 | 0.5686 | 0.4691 |
61
- | No log | 2.0 | 5 | 0.5967 | 0.6863 | 0.7053 | 0.6863 | 0.6139 |
62
- | No log | 2.8 | 7 | 0.5384 | 0.7843 | 0.7801 | 0.7843 | 0.7792 |
63
- | No log | 4.0 | 10 | 0.6429 | 0.6078 | 0.6547 | 0.6078 | 0.6164 |
64
- | No log | 4.8 | 12 | 0.6321 | 0.7255 | 0.7205 | 0.7255 | 0.7011 |
65
- | No log | 6.0 | 15 | 0.6473 | 0.7255 | 0.7164 | 0.7255 | 0.7095 |
66
- | No log | 6.8 | 17 | 0.7575 | 0.6863 | 0.6694 | 0.6863 | 0.6584 |
67
- | No log | 8.0 | 20 | 0.9926 | 0.7255 | 0.7312 | 0.7255 | 0.6908 |
68
- | No log | 8.8 | 22 | 0.9139 | 0.7255 | 0.7205 | 0.7255 | 0.7011 |
69
- | No log | 10.0 | 25 | 1.0884 | 0.7059 | 0.6937 | 0.7059 | 0.6845 |
70
- | No log | 10.8 | 27 | 1.2796 | 0.7451 | 0.7521 | 0.7451 | 0.7179 |
71
- | 0.287 | 12.0 | 30 | 1.3326 | 0.6863 | 0.6704 | 0.6863 | 0.6680 |
72
- | 0.287 | 12.8 | 32 | 1.5649 | 0.7255 | 0.7205 | 0.7255 | 0.7011 |
73
- | 0.287 | 14.0 | 35 | 1.7452 | 0.7255 | 0.7205 | 0.7255 | 0.7011 |
74
- | 0.287 | 14.8 | 37 | 1.7826 | 0.7255 | 0.7205 | 0.7255 | 0.7011 |
75
- | 0.287 | 16.0 | 40 | 1.9538 | 0.7255 | 0.7312 | 0.7255 | 0.6908 |
76
- | 0.287 | 16.8 | 42 | 1.8850 | 0.6863 | 0.6694 | 0.6863 | 0.6584 |
77
- | 0.287 | 18.0 | 45 | 1.7633 | 0.6863 | 0.6739 | 0.6863 | 0.6756 |
78
- | 0.287 | 18.8 | 47 | 1.7925 | 0.7059 | 0.6940 | 0.7059 | 0.6925 |
79
- | 0.287 | 20.0 | 50 | 2.1156 | 0.7255 | 0.7312 | 0.7255 | 0.6908 |
80
- | 0.287 | 20.8 | 52 | 2.0156 | 0.7255 | 0.7205 | 0.7255 | 0.7011 |
81
- | 0.287 | 22.0 | 55 | 1.8471 | 0.7255 | 0.7164 | 0.7255 | 0.7095 |
82
- | 0.287 | 22.8 | 57 | 1.7831 | 0.7647 | 0.7593 | 0.7647 | 0.7567 |
83
- | 0.0041 | 24.0 | 60 | 1.7628 | 0.7647 | 0.7593 | 0.7647 | 0.7567 |
84
- | 0.0041 | 24.8 | 62 | 1.8077 | 0.7451 | 0.7382 | 0.7451 | 0.7335 |
85
- | 0.0041 | 26.0 | 65 | 1.8068 | 0.7843 | 0.7823 | 0.7843 | 0.7745 |
86
- | 0.0041 | 26.8 | 67 | 1.7925 | 0.7647 | 0.7593 | 0.7647 | 0.7567 |
87
- | 0.0041 | 28.0 | 70 | 1.7721 | 0.7843 | 0.7823 | 0.7843 | 0.7745 |
88
- | 0.0041 | 28.8 | 72 | 1.7919 | 0.7647 | 0.7624 | 0.7647 | 0.7510 |
89
- | 0.0041 | 30.0 | 75 | 1.9588 | 0.7451 | 0.7521 | 0.7451 | 0.7179 |
90
- | 0.0041 | 30.8 | 77 | 1.9200 | 0.7451 | 0.7521 | 0.7451 | 0.7179 |
91
- | 0.0041 | 32.0 | 80 | 1.7746 | 0.7451 | 0.7521 | 0.7451 | 0.7179 |
92
- | 0.0041 | 32.8 | 82 | 1.7253 | 0.7647 | 0.7624 | 0.7647 | 0.7510 |
93
- | 0.0041 | 34.0 | 85 | 1.6992 | 0.7451 | 0.7382 | 0.7451 | 0.7335 |
94
- | 0.0041 | 34.8 | 87 | 1.6938 | 0.7451 | 0.7382 | 0.7451 | 0.7335 |
95
- | 0.0031 | 36.0 | 90 | 1.7014 | 0.7451 | 0.7382 | 0.7451 | 0.7335 |
96
 
97
 
98
  ### Framework versions
99
 
100
- - Transformers 4.38.2
101
  - Pytorch 2.2.1+cu121
102
- - Datasets 2.18.0
103
- - Tokenizers 0.15.2
 
19
 
20
  This model is a fine-tuned version of [microsoft/beit-base-patch16-224](https://huggingface.co/microsoft/beit-base-patch16-224) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 0.8528
23
+ - Accuracy: 0.8268
24
+ - Precision: 0.8303
25
+ - Recall: 0.8268
26
+ - F1 Score: 0.8283
27
 
28
  ## Model description
29
 
 
57
 
58
  | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1 Score |
59
  |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:--------:|
60
+ | No log | 0.8 | 2 | 0.6993 | 0.5882 | 0.5390 | 0.5882 | 0.5541 |
61
+ | No log | 2.0 | 5 | 0.5971 | 0.6863 | 0.6806 | 0.6863 | 0.6033 |
62
+ | No log | 2.8 | 7 | 0.5306 | 0.8039 | 0.8000 | 0.8039 | 0.8006 |
63
+ | No log | 4.0 | 10 | 0.4828 | 0.7255 | 0.7229 | 0.7255 | 0.6859 |
64
+ | No log | 4.8 | 12 | 0.3812 | 0.7843 | 0.7786 | 0.7843 | 0.7784 |
65
+ | 0.5413 | 6.0 | 15 | 0.5268 | 0.7451 | 0.7461 | 0.7451 | 0.7141 |
66
+ | 0.5413 | 6.8 | 17 | 0.5349 | 0.7451 | 0.8556 | 0.7451 | 0.7502 |
67
+ | 0.5413 | 8.0 | 20 | 0.4120 | 0.8039 | 0.8485 | 0.8039 | 0.7756 |
68
+ | 0.5413 | 8.8 | 22 | 0.3156 | 0.8039 | 0.8003 | 0.8039 | 0.7963 |
69
+ | 0.5413 | 10.0 | 25 | 0.3217 | 0.8039 | 0.8061 | 0.8039 | 0.7909 |
70
+ | 0.5413 | 10.8 | 27 | 0.5161 | 0.7843 | 0.7870 | 0.7843 | 0.7664 |
71
+ | 0.0919 | 12.0 | 30 | 0.3677 | 0.8431 | 0.8498 | 0.8431 | 0.8451 |
72
+ | 0.0919 | 12.8 | 32 | 0.4631 | 0.8431 | 0.8407 | 0.8431 | 0.8405 |
73
+ | 0.0919 | 14.0 | 35 | 0.5001 | 0.8235 | 0.8214 | 0.8235 | 0.8221 |
74
+ | 0.0919 | 14.8 | 37 | 0.4489 | 0.8431 | 0.8431 | 0.8431 | 0.8431 |
75
+ | 0.0919 | 16.0 | 40 | 0.5892 | 0.7843 | 0.7799 | 0.7843 | 0.7731 |
76
+ | 0.0919 | 16.8 | 42 | 0.6579 | 0.7843 | 0.7799 | 0.7843 | 0.7731 |
77
+ | 0.006 | 18.0 | 45 | 0.7038 | 0.7843 | 0.7799 | 0.7843 | 0.7731 |
78
+ | 0.006 | 18.8 | 47 | 0.5864 | 0.8627 | 0.8737 | 0.8627 | 0.8651 |
79
+ | 0.006 | 20.0 | 50 | 0.5488 | 0.8627 | 0.8737 | 0.8627 | 0.8651 |
80
+ | 0.006 | 20.8 | 52 | 0.6651 | 0.8039 | 0.8003 | 0.8039 | 0.7963 |
81
+ | 0.006 | 22.0 | 55 | 0.6265 | 0.8039 | 0.8000 | 0.8039 | 0.8006 |
82
+ | 0.006 | 22.8 | 57 | 0.5229 | 0.8627 | 0.8653 | 0.8627 | 0.8637 |
83
+ | 0.0048 | 24.0 | 60 | 0.5421 | 0.8627 | 0.8653 | 0.8627 | 0.8637 |
84
+ | 0.0048 | 24.8 | 62 | 0.6335 | 0.8235 | 0.8205 | 0.8235 | 0.8187 |
85
+ | 0.0048 | 26.0 | 65 | 1.0379 | 0.8039 | 0.8201 | 0.8039 | 0.7841 |
86
+ | 0.0048 | 26.8 | 67 | 0.9758 | 0.8235 | 0.8366 | 0.8235 | 0.8089 |
87
+ | 0.0048 | 28.0 | 70 | 0.6117 | 0.8235 | 0.8205 | 0.8235 | 0.8187 |
88
+ | 0.0048 | 28.8 | 72 | 0.5403 | 0.8627 | 0.8613 | 0.8627 | 0.8617 |
89
+ | 0.0063 | 30.0 | 75 | 0.6469 | 0.8431 | 0.8407 | 0.8431 | 0.8405 |
90
+ | 0.0063 | 30.8 | 77 | 0.7014 | 0.8235 | 0.8205 | 0.8235 | 0.8187 |
91
+ | 0.0063 | 32.0 | 80 | 0.7514 | 0.8235 | 0.8205 | 0.8235 | 0.8187 |
92
+ | 0.0063 | 32.8 | 82 | 0.7771 | 0.8235 | 0.8248 | 0.8235 | 0.8144 |
93
+ | 0.0063 | 34.0 | 85 | 0.7599 | 0.8039 | 0.8003 | 0.8039 | 0.7963 |
94
+ | 0.0063 | 34.8 | 87 | 0.7554 | 0.8039 | 0.8003 | 0.8039 | 0.7963 |
95
+ | 0.0045 | 36.0 | 90 | 0.7308 | 0.8039 | 0.8003 | 0.8039 | 0.7963 |
96
 
97
 
98
  ### Framework versions
99
 
100
+ - Transformers 4.40.1
101
  - Pytorch 2.2.1+cu121
102
+ - Datasets 2.19.0
103
+ - Tokenizers 0.19.1
config.json CHANGED
@@ -14,15 +14,15 @@
14
  "hidden_dropout_prob": 0.0,
15
  "hidden_size": 768,
16
  "id2label": {
17
- "0": "Absent",
18
- "1": "Present"
19
  },
20
  "image_size": 224,
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
  "label2id": {
24
- "Absent": 0,
25
- "Present": 1
26
  },
27
  "layer_norm_eps": 1e-12,
28
  "layer_scale_init_value": 0.1,
 
14
  "hidden_dropout_prob": 0.0,
15
  "hidden_size": 768,
16
  "id2label": {
17
+ "0": "Benign",
18
+ "1": "Malignant"
19
  },
20
  "image_size": 224,
21
  "initializer_range": 0.02,
22
  "intermediate_size": 3072,
23
  "label2id": {
24
+ "Benign": 0,
25
+ "Malignant": 1
26
  },
27
  "layer_norm_eps": 1e-12,
28
  "layer_scale_init_value": 0.1,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4cf1f884c4d7960adf228cffc9ce6c846fcf4438aa943bb4e949080c2aebd92c
3
  size 343080328
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:070c5b244ea67c14e33049ce84f0a492470a5f8f7b7bcefcb07dc7846bf3c7d3
3
  size 343080328
runs/May04_13-42-18_16ca11c6137e/events.out.tfevents.1714830139.16ca11c6137e.960.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9a6329899f9d0d70d38d672753c92b651e2dcda9b3c40be967a563536644c191
3
- size 22972
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57580ffae8fd477338a41af3cac83e4e2ae8b85457b177e7ece3b2cd92c1a280
3
+ size 23996
runs/May04_13-42-18_16ca11c6137e/events.out.tfevents.1714831102.16ca11c6137e.960.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b3e1eae5afbf79f9fa2204d436b477d28092060c28145da9d35243e86b61f7e
3
+ size 557
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2f84ad3380b312710e9817387d47973045f268d1a4130faf2df2cc0c2c171617
3
  size 4984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d9c4f61894c45d65bf229deb2fd4dc876cbd903f9e7da4c19cfe4a3825e08c68
3
  size 4984