ErnestBeckham
commited on
Commit
·
ca708c2
1
Parent(s):
ea9a32f
updated readme.md
Browse files
README.md
CHANGED
@@ -1,10 +1,22 @@
|
|
1 |
---
|
2 |
library_name: keras
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
4 |
|
5 |
## Model description
|
6 |
|
7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
8 |
|
9 |
## Intended uses & limitations
|
10 |
|
@@ -12,7 +24,20 @@ More information needed
|
|
12 |
|
13 |
## Training and evaluation data
|
14 |
|
15 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
|
17 |
## Training procedure
|
18 |
|
|
|
1 |
---
|
2 |
library_name: keras
|
3 |
+
license: mit
|
4 |
+
metrics:
|
5 |
+
- accuracy
|
6 |
+
tags:
|
7 |
+
- biology
|
8 |
---
|
9 |
|
10 |
## Model description
|
11 |
|
12 |
+
This is Vision Transformer model trained for cancer classification. To make single model to predict any cancer, I trained this ViT model. following are the cancer types that model can predict:
|
13 |
+
* Brain cancer
|
14 |
+
* Breast Cancer (histopathology)
|
15 |
+
* Lung & Colon Cancer (histopathology)
|
16 |
+
* Cervical Caner
|
17 |
+
* Kidney Cancer
|
18 |
+
* Lymphoma
|
19 |
+
* Oral
|
20 |
|
21 |
## Intended uses & limitations
|
22 |
|
|
|
24 |
|
25 |
## Training and evaluation data
|
26 |
|
27 |
+
Confusion matrix and classification report attached below:
|
28 |
+
<details>
|
29 |
+
<summary>Confusion Matrix</summary>
|
30 |
+
|
31 |
+
![Model Image](./model.png)
|
32 |
+
|
33 |
+
</details>
|
34 |
+
|
35 |
+
<details>
|
36 |
+
<summary>Classification Report</summary>
|
37 |
+
|
38 |
+
![Model Image](./model.png)
|
39 |
+
|
40 |
+
</details>
|
41 |
|
42 |
## Training procedure
|
43 |
|