Downloads · 30 days
5
6% of all-time downloads
akar49/VIT_braintumor_classifier
VIT_braintumor_classifier is a image classification model from akar49. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
probably proofread and complete it, then remove this comment. --
Downloads · 30 days
5
6% of all-time downloads
All-time downloads
79
Public
Repo size
343 MB
Likes
0
Public
Click a slice to open those files.
.h5343 MB · 100%
From the Hugging Face model README
This model is a fine-tuned version of google/vit-base-patch16-224-in21k on an unknown dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Train Loss | Validation Loss | Train Accuracy | Epoch |
|---|---|---|---|
| 0.7946 | 1.1484 | 0.6272 | 0 |
| 0.3246 | 1.1792 | 0.6769 | 1 |
| 0.2266 | 1.2812 | 0.6842 | 2 |
| 0.1841 | 1.5085 | 0.6754 | 3 |
| 0.1589 | 1.4224 | 0.6944 | 4 |
| 0.1244 | 1.4229 | 0.6901 | 5 |
| 0.1174 | 1.4858 | 0.6784 | 6 |
| 0.1133 | 1.4221 | 0.6974 | 7 |
| 0.1026 | 1.4273 | 0.7003 | 8 |
| 0.1083 | 1.5406 | 0.7003 | 9 |
| 0.1038 | 1.6223 | 0.6974 | 10 |
| 0.0876 | 1.5613 | 0.6959 | 11 |
| 0.1018 | 1.4540 | 0.7149 | 12 |
| 0.0808 | 1.4853 | 0.7193 | 13 |
| 0.0945 | 1.7241 | 0.6974 | 14 |