Update README.md

README.md

@@ -2,10 +2,31 @@
 tags:
 - image-classification
 - pytorch
+- medical
+- colon
 metrics:
 - f1
 ---
 
 # Vision Transformer fine-tuned on kvasir_v2 for colonoscopy classification
 
+## Metrics
+
+```
+                        precision    recall  f1-score   support
+
+    dyed-lifted-polyps       0.95      0.93      0.94        60
+dyed-resection-margins       0.97      0.95      0.96        64
+           esophagitis       0.93      0.79      0.85        67
+          normal-cecum       1.00      0.98      0.99        54
+        normal-pylorus       0.95      1.00      0.97        57
+         normal-z-line       0.82      0.93      0.87        67
+                polyps       0.92      0.92      0.92        52
+    ulcerative-colitis       0.93      0.95      0.94        59
+
+              accuracy                           0.93       480
+             macro avg       0.93      0.93      0.93       480
+          weighted avg       0.93      0.93      0.93       480
+```
+
 > Disclaimer: This model was trained for research only