|
--- |
|
license: apache-2.0 |
|
metrics: |
|
- accuracy |
|
- f1 |
|
--- |
|
Returns a common mammal type given an image. |
|
|
|
See https://www.kaggle.com/code/dima806/mammals-45-types-image-classification-vit for more details. |
|
|
|
``` |
|
Classification report: |
|
|
|
precision recall f1-score support |
|
|
|
african_elephant 1.0000 1.0000 1.0000 71 |
|
alpaca 0.9200 0.9718 0.9452 71 |
|
american_bison 1.0000 1.0000 1.0000 71 |
|
anteater 0.9853 0.9437 0.9640 71 |
|
arctic_fox 0.9286 0.9155 0.9220 71 |
|
armadillo 0.9726 1.0000 0.9861 71 |
|
baboon 0.9718 0.9718 0.9718 71 |
|
badger 1.0000 0.9718 0.9857 71 |
|
blue_whale 0.9710 0.9437 0.9571 71 |
|
brown_bear 0.9722 0.9859 0.9790 71 |
|
camel 0.9861 1.0000 0.9930 71 |
|
dolphin 0.8974 0.9859 0.9396 71 |
|
giraffe 0.9857 0.9718 0.9787 71 |
|
groundhog 0.9714 0.9577 0.9645 71 |
|
highland_cattle 0.9859 0.9859 0.9859 71 |
|
horse 1.0000 0.9859 0.9929 71 |
|
jackal 0.9577 0.9444 0.9510 72 |
|
kangaroo 0.8415 0.9583 0.8961 72 |
|
koala 0.9589 0.9859 0.9722 71 |
|
manatee 0.9861 0.9861 0.9861 72 |
|
mongoose 0.9483 0.7746 0.8527 71 |
|
mountain_goat 0.9855 0.9577 0.9714 71 |
|
opossum 1.0000 0.9577 0.9784 71 |
|
orangutan 1.0000 1.0000 1.0000 71 |
|
otter 1.0000 0.9577 0.9784 71 |
|
polar_bear 0.9706 0.9296 0.9496 71 |
|
porcupine 1.0000 0.9722 0.9859 72 |
|
red_panda 0.9718 0.9718 0.9718 71 |
|
rhinoceros 0.9859 0.9859 0.9859 71 |
|
sea_lion 0.7600 0.8028 0.7808 71 |
|
seal 0.8308 0.7500 0.7883 72 |
|
snow_leopard 1.0000 1.0000 1.0000 71 |
|
squirrel 0.9444 0.9577 0.9510 71 |
|
sugar_glider 0.8554 1.0000 0.9221 71 |
|
tapir 1.0000 1.0000 1.0000 71 |
|
vampire_bat 1.0000 0.9861 0.9930 72 |
|
vicuna 1.0000 0.8873 0.9403 71 |
|
walrus 0.9342 0.9861 0.9595 72 |
|
warthog 0.9571 0.9437 0.9504 71 |
|
water_buffalo 0.9333 0.9859 0.9589 71 |
|
weasel 0.9583 0.9583 0.9583 72 |
|
wildebeest 0.9577 0.9444 0.9510 72 |
|
wombat 0.8947 0.9577 0.9252 71 |
|
yak 1.0000 0.9437 0.9710 71 |
|
zebra 0.9595 1.0000 0.9793 71 |
|
|
|
accuracy 0.9572 3204 |
|
macro avg 0.9587 0.9573 0.9572 3204 |
|
weighted avg 0.9586 0.9572 0.9572 3204 |
|
``` |