|
--- |
|
library_name: keras |
|
license: mit |
|
metrics: |
|
- accuracy |
|
pipeline_tag: image-classification |
|
--- |
|
|
|
## Model description |
|
|
|
This mode for classify the lung cancer using histopathology images. This is Transformer based model (ViT). |
|
|
|
## Intended uses & limitations |
|
|
|
More information needed |
|
|
|
## Training and evaluation data |
|
|
|
Dataset Source: https://www.kaggle.com/datasets/scipygaurav/lung-and-colon-cancer-dataset-splitted |
|
|
|
## 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 Acccuracy |
|
<details> |
|
<summary>Confusion Matrix</summary> |
|
|
|
 |
|
 |
|
</details> |
|
|
|
## Model Plot |
|
|
|
<details> |
|
<summary>View Model Plot</summary> |
|
|
|
 |
|
|
|
</details> |
