chexnet-multilabel-14 / config.json
alex17cmbs's picture
Upload densenet121_chexnet multi-label model (AUC: 0.7486)
e0ea27a verified
{
"model_type": "densenet121_chexnet",
"model_name": "densenet121",
"num_classes": 14,
"class_names": [
"Atelectasis",
"Cardiomegaly",
"Effusion",
"Infiltration",
"Mass",
"Nodule",
"Pneumonia",
"Pneumothorax",
"Consolidation",
"Edema",
"Emphysema",
"Fibrosis",
"Pleural_Thickening",
"Hernia"
],
"image_size": 224,
"task": "multi-label-classification",
"dataset": "NIH Chest X-ray (ChestX-ray14)",
"training_config": {
"data_variant": "full",
"batch_size": 16,
"image_size": 224,
"num_classes": 14,
"learning_rate": 0.001,
"num_epochs": 50,
"scheduler": "plateau (factor=0.5, patience=5)",
"optimizer": "Adam (betas=(0.9, 0.999))",
"loss": "BCEWithLogitsLoss (non pond\u00e9r\u00e9e)"
},
"metrics": {
"auc_macro": 0.7485587174542548,
"auc_micro": 0.821897745043009,
"auc_weighted": 0.7379569893821755,
"f1_macro": 0.04718208678601691,
"f1_micro": 0.09423200243005164,
"f1_weighted": 0.08346628598901618,
"precision_macro": 0.32550546293388616,
"precision_micro": 0.5761452744531572,
"recall_macro": 0.028426153207235426,
"recall_micro": 0.051312210541792254,
"mAP": 0.1908223526435699
},
"uploaded_at": "2025-12-06T23:55:13.239563"
}