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