Downloads · 30 days
8
10% of all-time downloads
vit-chest-xray-multilabel
alex17cmbs/vit-chest-xray-multilabel
vit-chest-xray-multilabel is a image classification model from alex17cmbs. Use it when you need a label for an image. It is set up for pytorch. The card lists the license as apache-2.0.
This model is a vit-base-15classes trained on the NIH Chest X-ray dataset for multi-label classification of 14 pathologies.
All-time downloads
79
Public
Repo size
687 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin343 MB · 100%
At a glance
- Task
- Image Classification
- Library
- pytorch
- License
- apache-2.0
- Access
- Public
- Created
- Jan 4, 2026
- Updated
- Jan 4, 2026
- SHA
- de9925c5
Model card
From the Hugging Face model README
vit-base-15classes
This model is a vit-base-15classes trained on the NIH Chest X-ray dataset for multi-label classification of 14 pathologies.
Metrics
- auc_macro: 0.7464
- auc_micro: 0.8365
- mAP: 0.2165
Training Configuration
- learning_rate: 0.0001
- batch_size: 16
- num_epochs: 50
Usage
import torch
# Load model logic here
- Task
- Image Classification
- Library
- pytorch
- License
- apache-2.0
- Languages
- en
- Created
- Jan 4, 2026
- Updated
- Jan 4, 2026