Update README.md

README.md

@@ -3,4 +3,25 @@ license: apache-2.0
 metrics:
 - accuracy
 ---
-See https://www.kaggle.com/code/dima806/diamond-types-image-detection-vit for details.
+Returns diamond type given an image.
+
+See https://www.kaggle.com/code/dima806/diamond-types-image-detection-vit for details.
+
+```
+Classification report:
+
+              precision    recall  f1-score   support
+
+        pear     1.0000    0.9956    0.9978      2280
+     cushion     0.9418    1.0000    0.9700      2280
+    princess     0.9996    0.9890    0.9943      2279
+       round     0.9964    0.9781    0.9872      2279
+    marquise     0.9987    0.9882    0.9934      2279
+        oval     0.9996    0.9904    0.9949      2280
+     emerald     1.0000    0.9956    0.9978      2279
+       heart     0.9987    0.9943    0.9965      2280
+
+    accuracy                         0.9914     18236
+   macro avg     0.9918    0.9914    0.9915     18236
+weighted avg     0.9918    0.9914    0.9915     18236
+```