Update README.md

README.md

@@ -4,4 +4,62 @@ metrics:
 - accuracy
 - f1
 ---
-See https://www.kaggle.com/code/dima806/mammals-45-types-image-classification-vit for more details.
+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
+```