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