|
--- |
|
license: apache-2.0 |
|
metrics: |
|
- accuracy |
|
- f1 |
|
--- |
|
Returns pokemon type given an image. |
|
|
|
See https://www.kaggle.com/code/dima806/pokemon-common-types-image-detection-vit for more details. |
|
|
|
``` |
|
Accuracy: 0.9588 |
|
F1 Score: 0.9459 |
|
|
|
Classification report: |
|
|
|
precision recall f1-score support |
|
|
|
Wartortle 0.9615 0.9615 0.9615 26 |
|
Arcanine 1.0000 1.0000 1.0000 27 |
|
Staryu 1.0000 1.0000 1.0000 27 |
|
Arbok 1.0000 1.0000 1.0000 26 |
|
Butterfree 0.0000 0.0000 0.0000 26 |
|
Geodude 1.0000 1.0000 1.0000 27 |
|
Seaking 1.0000 1.0000 1.0000 26 |
|
Diglett 1.0000 1.0000 1.0000 27 |
|
Jynx 1.0000 1.0000 1.0000 26 |
|
Sandslash 0.9286 1.0000 0.9630 26 |
|
Magneton 1.0000 1.0000 1.0000 27 |
|
Scyther 1.0000 1.0000 1.0000 27 |
|
Kabuto 1.0000 1.0000 1.0000 26 |
|
Cubone 0.8276 0.9231 0.8727 26 |
|
Golem 1.0000 1.0000 1.0000 26 |
|
Dewgong 0.9630 1.0000 0.9811 26 |
|
Pidgey 1.0000 0.9259 0.9615 27 |
|
Kadabra 0.5200 1.0000 0.6842 26 |
|
Ditto 1.0000 1.0000 1.0000 26 |
|
Venomoth 0.5400 1.0000 0.7013 27 |
|
Rattata 1.0000 1.0000 1.0000 27 |
|
Alakazam 0.0000 0.0000 0.0000 26 |
|
Machoke 1.0000 0.9615 0.9804 26 |
|
Farfetchd 1.0000 1.0000 1.0000 27 |
|
Omastar 1.0000 0.9615 0.9804 26 |
|
Machamp 0.9630 1.0000 0.9811 26 |
|
Jigglypuff 1.0000 1.0000 1.0000 27 |
|
Dragonite 1.0000 1.0000 1.0000 26 |
|
Weepinbell 1.0000 1.0000 1.0000 26 |
|
Sandshrew 1.0000 1.0000 1.0000 26 |
|
Dugtrio 1.0000 1.0000 1.0000 27 |
|
Mankey 0.8276 0.8889 0.8571 27 |
|
Hitmonchan 0.8667 1.0000 0.9286 26 |
|
Spearow 1.0000 1.0000 1.0000 26 |
|
Caterpie 1.0000 1.0000 1.0000 27 |
|
Dratini 0.0000 0.0000 0.0000 26 |
|
Bulbasaur 1.0000 1.0000 1.0000 26 |
|
Tentacool 1.0000 1.0000 1.0000 26 |
|
Gengar 1.0000 1.0000 1.0000 26 |
|
Machop 0.9643 1.0000 0.9818 27 |
|
Raichu 1.0000 1.0000 1.0000 26 |
|
Alolan Sandslash 0.0000 0.0000 0.0000 26 |
|
Eevee 1.0000 1.0000 1.0000 27 |
|
Abra 1.0000 1.0000 1.0000 27 |
|
Haunter 1.0000 1.0000 1.0000 27 |
|
Metapod 1.0000 1.0000 1.0000 27 |
|
Fearow 0.9630 1.0000 0.9811 26 |
|
Nidorina 0.8966 1.0000 0.9455 26 |
|
Zapdos 1.0000 1.0000 1.0000 27 |
|
Ninetales 1.0000 0.9630 0.9811 27 |
|
Chansey 1.0000 1.0000 1.0000 27 |
|
Kangaskhan 0.9630 1.0000 0.9811 26 |
|
Poliwrath 1.0000 0.9630 0.9811 27 |
|
Gyarados 1.0000 1.0000 1.0000 27 |
|
Charmeleon 1.0000 1.0000 1.0000 26 |
|
Vulpix 1.0000 1.0000 1.0000 26 |
|
Pidgeot 1.0000 0.8846 0.9388 26 |
|
Blastoise 0.9630 1.0000 0.9811 26 |
|
Porygon 1.0000 1.0000 1.0000 26 |
|
Psyduck 0.9643 1.0000 0.9818 27 |
|
Dragonair 0.5400 1.0000 0.7013 27 |
|
Raticate 0.9630 1.0000 0.9811 26 |
|
Squirtle 1.0000 0.9615 0.9804 26 |
|
Charizard 1.0000 1.0000 1.0000 26 |
|
Electrode 1.0000 0.9615 0.9804 26 |
|
Flareon 1.0000 1.0000 1.0000 26 |
|
Exeggutor 0.9643 1.0000 0.9818 27 |
|
Pikachu 1.0000 1.0000 1.0000 26 |
|
Wigglytuff 1.0000 1.0000 1.0000 26 |
|
Venusaur 1.0000 0.9615 0.9804 26 |
|
Mewtwo 1.0000 1.0000 1.0000 26 |
|
Clefable 1.0000 1.0000 1.0000 27 |
|
Oddish 1.0000 1.0000 1.0000 26 |
|
Ekans 1.0000 1.0000 1.0000 26 |
|
Shellder 1.0000 1.0000 1.0000 26 |
|
Marowak 0.9130 0.8077 0.8571 26 |
|
Kakuna 1.0000 1.0000 1.0000 27 |
|
Rapidash 1.0000 0.9615 0.9804 26 |
|
Rhydon 1.0000 0.9630 0.9811 27 |
|
Ivysaur 1.0000 1.0000 1.0000 26 |
|
Slowpoke 1.0000 1.0000 1.0000 26 |
|
Lapras 1.0000 1.0000 1.0000 27 |
|
Clefairy 1.0000 1.0000 1.0000 26 |
|
Hitmonlee 1.0000 1.0000 1.0000 26 |
|
Jolteon 1.0000 1.0000 1.0000 26 |
|
Growlithe 1.0000 1.0000 1.0000 27 |
|
Gastly 1.0000 1.0000 1.0000 27 |
|
Aerodactyl 1.0000 1.0000 1.0000 27 |
|
Weedle 1.0000 1.0000 1.0000 26 |
|
Tauros 1.0000 1.0000 1.0000 27 |
|
Seel 0.8929 0.9615 0.9259 26 |
|
Zubat 1.0000 1.0000 1.0000 26 |
|
Meowth 0.0000 0.0000 0.0000 26 |
|
Persian 0.6341 1.0000 0.7761 26 |
|
Articuno 0.9310 1.0000 0.9643 27 |
|
Weezing 0.9643 1.0000 0.9818 27 |
|
Magnemite 1.0000 1.0000 1.0000 27 |
|
Omanyte 0.9630 1.0000 0.9811 26 |
|
Mew 1.0000 1.0000 1.0000 26 |
|
Vileplume 1.0000 1.0000 1.0000 27 |
|
Nidoqueen 0.9615 0.9259 0.9434 27 |
|
Vaporeon 0.9000 1.0000 0.9474 27 |
|
Ponyta 0.9630 1.0000 0.9811 26 |
|
Moltres 1.0000 1.0000 1.0000 27 |
|
Voltorb 0.9630 1.0000 0.9811 26 |
|
Magikarp 1.0000 1.0000 1.0000 27 |
|
Beedrill 1.0000 1.0000 1.0000 26 |
|
Nidoking 1.0000 1.0000 1.0000 27 |
|
Paras 1.0000 1.0000 1.0000 26 |
|
Grimer 1.0000 0.9615 0.9804 26 |
|
Dodrio 1.0000 1.0000 1.0000 26 |
|
Charmander 1.0000 1.0000 1.0000 26 |
|
Muk 1.0000 0.9615 0.9804 26 |
|
Primeape 0.8966 0.9630 0.9286 27 |
|
Victreebel 1.0000 1.0000 1.0000 26 |
|
Golbat 1.0000 1.0000 1.0000 26 |
|
Horsea 1.0000 1.0000 1.0000 27 |
|
Goldeen 1.0000 1.0000 1.0000 27 |
|
Pidgeotto 0.8966 1.0000 0.9455 26 |
|
Koffing 0.9630 1.0000 0.9811 26 |
|
Seadra 0.5870 1.0000 0.7397 27 |
|
Tentacruel 1.0000 1.0000 1.0000 26 |
|
Pinsir 1.0000 1.0000 1.0000 26 |
|
Cloyster 1.0000 1.0000 1.0000 26 |
|
Gloom 1.0000 1.0000 1.0000 26 |
|
Graveler 1.0000 1.0000 1.0000 26 |
|
Magmar 1.0000 1.0000 1.0000 27 |
|
Krabby 0.9286 1.0000 0.9630 26 |
|
Electabuzz 1.0000 1.0000 1.0000 27 |
|
Poliwhirl 0.9643 1.0000 0.9818 27 |
|
Golduck 0.9310 1.0000 0.9643 27 |
|
Onix 1.0000 1.0000 1.0000 27 |
|
Nidorino 1.0000 1.0000 1.0000 27 |
|
Snorlax 0.9630 1.0000 0.9811 26 |
|
Starmie 1.0000 1.0000 1.0000 27 |
|
Slowbro 1.0000 1.0000 1.0000 26 |
|
MrMime 1.0000 1.0000 1.0000 26 |
|
Venonat 1.0000 1.0000 1.0000 27 |
|
Kabutops 1.0000 1.0000 1.0000 26 |
|
Drowzee 1.0000 1.0000 1.0000 26 |
|
Rhyhorn 1.0000 1.0000 1.0000 26 |
|
Tangela 1.0000 1.0000 1.0000 27 |
|
Doduo 1.0000 1.0000 1.0000 27 |
|
Exeggcute 1.0000 1.0000 1.0000 26 |
|
Poliwag 1.0000 1.0000 1.0000 27 |
|
Lickitung 1.0000 1.0000 1.0000 26 |
|
Hypno 0.9286 1.0000 0.9630 26 |
|
Bellsprout 1.0000 1.0000 1.0000 27 |
|
Parasect 1.0000 1.0000 1.0000 26 |
|
Kingler 1.0000 0.9231 0.9600 26 |
|
|
|
accuracy 0.9588 3960 |
|
macro avg 0.9382 0.9583 0.9459 3960 |
|
weighted avg 0.9386 0.9588 0.9463 3960 |
|
``` |