ViT-Lungs / README.md
ErnestBeckham's picture
updated
a5c8998
metadata
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

Confusion Matrix

Model Image Model Image

Model Plot

View Model Plot

Model Image