---
license: apache-2.0
metrics:
- accuracy
---
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
```