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:
Confusion Matrix
Classification Report
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
View Model Plot
- Downloads last month
- 9