| 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 | |
| ```python | |
| import torch | |
| # Load model logic here | |
| ``` | |