metadata
language: en
license: apache-2.0
library_name: pytorch
pipeline_tag: image-classification
tags:
- chexnet
- chest-xray
- multi-label
- 14-pathologies
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