Update README.md
Browse files
README.md
CHANGED
@@ -3,4 +3,25 @@ license: apache-2.0
|
|
3 |
metrics:
|
4 |
- accuracy
|
5 |
---
|
6 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
metrics:
|
4 |
- accuracy
|
5 |
---
|
6 |
+
Returns diamond type given an image.
|
7 |
+
|
8 |
+
See https://www.kaggle.com/code/dima806/diamond-types-image-detection-vit for details.
|
9 |
+
|
10 |
+
```
|
11 |
+
Classification report:
|
12 |
+
|
13 |
+
precision recall f1-score support
|
14 |
+
|
15 |
+
pear 1.0000 0.9956 0.9978 2280
|
16 |
+
cushion 0.9418 1.0000 0.9700 2280
|
17 |
+
princess 0.9996 0.9890 0.9943 2279
|
18 |
+
round 0.9964 0.9781 0.9872 2279
|
19 |
+
marquise 0.9987 0.9882 0.9934 2279
|
20 |
+
oval 0.9996 0.9904 0.9949 2280
|
21 |
+
emerald 1.0000 0.9956 0.9978 2279
|
22 |
+
heart 0.9987 0.9943 0.9965 2280
|
23 |
+
|
24 |
+
accuracy 0.9914 18236
|
25 |
+
macro avg 0.9918 0.9914 0.9915 18236
|
26 |
+
weighted avg 0.9918 0.9914 0.9915 18236
|
27 |
+
```
|