File size: 833 Bytes
dafe9bd 0e67933 dafe9bd 5044b1c 7869331 b27477d 7869331 5044b1c 7869331 b27477d 5044b1c b27477d 5044b1c b27477d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
---
license: apache-2.0
metrics:
- accuracy
base_model:
- google/vit-base-patch16-224-in21k
---
Achieved 78% weighted accuracy for classification between 3 common galaxy types (S, E, Sb).
See [my Kaggle notebook](https://www.kaggle.com/code/dima806/galaxy-type-image-detection-vit) for more details.
![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/g5Gx6ra0hLYDU6BujnhPJ.png)
```
Classification report:
precision recall f1-score support
E 0.7656 0.8848 0.8209 13592
S 0.7526 0.6685 0.7081 13591
SB 0.8262 0.7900 0.8077 13591
accuracy 0.7811 40774
macro avg 0.7815 0.7811 0.7789 40774
weighted avg 0.7815 0.7811 0.7789 40774
``` |