|
--- |
|
library_name: keras |
|
license: mit |
|
metrics: |
|
- accuracy |
|
- f1 |
|
- precision |
|
tags: |
|
- biology |
|
pipeline_tag: image-classification |
|
--- |
|
|
|
## Model description |
|
|
|
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: |
|
* Brain cancer |
|
* Breast Cancer (histopathology) |
|
* Lung & Colon Cancer (histopathology) |
|
* Cervical Caner |
|
* Kidney Cancer |
|
* Lymphoma |
|
* Oral |
|
|
|
## Intended uses & limitations |
|
|
|
More information needed |
|
|
|
## Training and evaluation data |
|
|
|
Confusion matrix and classification report attached below: |
|
<details> |
|
<summary>Confusion Matrix</summary> |
|
|
|
 |
|
|
|
</details> |
|
|
|
<details> |
|
<summary>Classification Report</summary> |
|
|
|
 |
|
|
|
</details> |
|
|
|
## Training procedure |
|
|
|
### Training hyperparameters |
|
|
|
The following hyperparameters were used during training: |
|
|
|
| Hyperparameters | Value | |
|
| :-- | :-- | |
|
| name | Adam | |
|
| learning_rate | 0.0001 | |
|
| decay | 0.0 | |
|
| beta_1 | 0.9 | |
|
| beta_2 | 0.999 | |
|
| epsilon | 1e-07 | |
|
| amsgrad | False | |
|
| training_precision | float32 | |
|
|
|
|
|
## Model Plot |
|
|
|
<details> |
|
<summary>View Model Plot</summary> |
|
|
|
 |
|
|
|
</details> |
