---
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>

  ![Model Image](./cm.png)
  
</details>

 ## Model Plot

<details>
<summary>View Model Plot</summary>

![Model Image](./model.png)

</details>