Downloads · 30 days
0
ViT-Lungs
ErnestBeckham/ViT-Lungs
ViT-Lungs is a image classification model from ErnestBeckham. Use it when you need a label for an image. It is set up for keras. The card lists the license as mit.
This mode for classify the lung cancer using histopathology images. This is Transformer based model (ViT).
Access
Public
Updated Nov 23, 2023
Repo size
291 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.data-00000-of-00001285 MB · 98%
At a glance
- Task
- Image Classification
- Library
- keras
- License
- mit
- Access
- Public
- Created
- Nov 23, 2023
- Updated
- Nov 23, 2023
- SHA
- a5c89980
Model card
From the Hugging Face model README
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 Plot
<details> <summary>View Model Plot</summary>
- Task
- Image Classification
- Library
- keras
- License
- mit
- Created
- Nov 23, 2023
- Updated
- Nov 23, 2023