densenet121-chexnet-14classes
This model is a densenet121-chexnet-14classes trained on the NIH Chest X-ray dataset for multi-label classification of 14 pathologies.
Metrics
- auc_macro: 0.7626
- auc_micro: 0.8496
- mAP: 0.2341
Training Configuration
- learning_rate: 0.001
- batch_size: 16
- num_epochs: 50
Usage
import torch
# Load model logic here
- Downloads last month
- 8