alex17cmbs commited on
Commit
83cd0c6
·
verified ·
1 Parent(s): 586b6a4

Upload densenet121-chexnet-14classes model (AUC: 0.7626)

Browse files
Files changed (3) hide show
  1. README.md +35 -0
  2. config.json +35 -0
  3. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: apache-2.0
4
+ library_name: pytorch
5
+ pipeline_tag: image-classification
6
+ tags:
7
+ - chexnet
8
+ - chest-xray
9
+ - multi-label
10
+ - 14-pathologies
11
+ ---
12
+
13
+ # densenet121-chexnet-14classes
14
+
15
+ This model is a densenet121-chexnet-14classes trained on the NIH Chest X-ray dataset for multi-label classification of 14 pathologies.
16
+
17
+
18
+ ## Metrics
19
+ - **auc_macro**: 0.7626
20
+ - **auc_micro**: 0.8496
21
+ - **mAP**: 0.2341
22
+
23
+
24
+ ## Training Configuration
25
+ - **learning_rate**: 0.001
26
+ - **batch_size**: 16
27
+ - **num_epochs**: 50
28
+
29
+
30
+ ## Usage
31
+
32
+ ```python
33
+ import torch
34
+ # Load model logic here
35
+ ```
config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "densenet121-chexnet-14classes",
3
+ "num_classes": 15,
4
+ "class_names": [
5
+ "No Finding",
6
+ "Atelectasis",
7
+ "Cardiomegaly",
8
+ "Effusion",
9
+ "Infiltration",
10
+ "Mass",
11
+ "Nodule",
12
+ "Pneumonia",
13
+ "Pneumothorax",
14
+ "Consolidation",
15
+ "Edema",
16
+ "Emphysema",
17
+ "Fibrosis",
18
+ "Pleural_Thickening",
19
+ "Hernia"
20
+ ],
21
+ "image_size": 224,
22
+ "task": "multi-label-classification",
23
+ "dataset": "NIH Chest X-ray",
24
+ "uploaded_at": "2026-01-04T09:35:34.631516",
25
+ "training_config": {
26
+ "learning_rate": 0.001,
27
+ "batch_size": 16,
28
+ "num_epochs": 50
29
+ },
30
+ "metrics": {
31
+ "auc_macro": 0.7626,
32
+ "auc_micro": 0.8496,
33
+ "mAP": 0.2341
34
+ }
35
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc1d8449838118bc3af061411271e18fc7d637a29a63f146150cd90eb95fc1e5
3
+ size 28486719