Update README.md

README.md

@@ -4,4 +4,27 @@ metrics:
 - accuracy
 - f1
 ---
-See https://www.kaggle.com/code/dima806/ship-type-detection-vit for more details.
+Returns ship type given an image.
+
+See https://www.kaggle.com/code/dima806/ship-type-detection-vit for more details.
+
+```
+Classification report:
+
+                  precision    recall  f1-score   support
+
+         Bulkers     0.9808    1.0000    0.9903       409
+    Recreational     0.9663    0.9805    0.9734       410
+        Sailboat     0.9975    0.9829    0.9901       409
+             DDG     0.9951    0.9927    0.9939       409
+  Container Ship     1.0000    0.9902    0.9951       409
+             Tug     0.9878    0.9878    0.9878       409
+Aircraft Carrier     1.0000    0.9878    0.9938       409
+          Cruise     0.9951    1.0000    0.9976       409
+       Submarine     0.9976    0.9976    0.9976       409
+     Car Carrier     0.9976    0.9976    0.9976       410
+
+        accuracy                         0.9917      4092
+       macro avg     0.9918    0.9917    0.9917      4092
+    weighted avg     0.9918    0.9917    0.9917      4092
+```