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
```