Update README.md
Browse files
README.md
CHANGED
|
@@ -5,6 +5,8 @@ tags:
|
|
| 5 |
- breast-cancer
|
| 6 |
- classification
|
| 7 |
- swin-transformer
|
|
|
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
# 🩺 Breast Cancer Classification Using Swin Transformer (Hugging Face)
|
|
@@ -111,4 +113,4 @@ with torch.no_grad():
|
|
| 111 |
predicted_class = outputs.logits.argmax(-1).item()
|
| 112 |
|
| 113 |
labels = ["benign", "malignant"]
|
| 114 |
-
print("Predicted class:", labels[predicted_class])
|
|
|
|
| 5 |
- breast-cancer
|
| 6 |
- classification
|
| 7 |
- swin-transformer
|
| 8 |
+
metrics:
|
| 9 |
+
- accuracy
|
| 10 |
---
|
| 11 |
|
| 12 |
# 🩺 Breast Cancer Classification Using Swin Transformer (Hugging Face)
|
|
|
|
| 113 |
predicted_class = outputs.logits.argmax(-1).item()
|
| 114 |
|
| 115 |
labels = ["benign", "malignant"]
|
| 116 |
+
print("Predicted class:", labels[predicted_class])
|