ErnestBeckham commited on
Commit
ca708c2
·
1 Parent(s): ea9a32f

updated readme.md

Browse files
Files changed (1) hide show
  1. README.md +27 -2
README.md CHANGED
@@ -1,10 +1,22 @@
1
  ---
2
  library_name: keras
 
 
 
 
 
3
  ---
4
 
5
  ## Model description
6
 
7
- More information needed
 
 
 
 
 
 
 
8
 
9
  ## Intended uses & limitations
10
 
@@ -12,7 +24,20 @@ More information needed
12
 
13
  ## Training and evaluation data
14
 
15
- More information needed
 
 
 
 
 
 
 
 
 
 
 
 
 
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