{
  "seed": 42,
  "data": {
    "dataset_name": "HlexNC/chest-xray-14-320",
    "image_size": 320,
    "num_workers": 4,
    "pin_memory": true,
    "train_split": 0.7,
    "val_split": 0.1,
    "test_split": 0.2,
    "augmentation": {
      "horizontal_flip": true,
      "rotation_degrees": 15,
      "affine_translate": 0.05,
      "affine_shear": 5,
      "color_jitter": {
        "brightness": 0.3,
        "contrast": 0.3,
        "saturation": 0.05
      },
      "gaussian_blur_prob": 0.2,
      "random_erasing_prob": 0.1
    },
    "labels": [
      "Atelectasis",
      "Cardiomegaly",
      "Effusion",
      "Infiltration",
      "Mass",
      "Nodule",
      "Pneumonia",
      "Pneumothorax",
      "Consolidation",
      "Edema",
      "Emphysema",
      "Fibrosis",
      "Pleural_Thickening",
      "Hernia"
    ],
    "data_dir": "/kaggle/working/data",
    "hf_dataset_repo": "HlexNC/chest-xray-14-320",
    "hf_dataset_revision": "44443e6ee968b3c6094b63f14a27698c40b50680"
  },
  "training": {
    "epochs": 60,
    "batch_size": 24,
    "learning_rate": 0.0001,
    "weight_decay": 0.0001,
    "optimizer": "adamw",
    "scheduler": "cosine",
    "warmup_epochs": 3,
    "early_stopping_patience": 15,
    "gradient_clip_norm": 1.0,
    "use_amp": true,
    "grad_accum_steps": 4,
    "multilabel_weight": 1.0,
    "binary_weight": 0.5
  },
  "evaluation": {
    "primary_metric": "auc_roc_macro",
    "threshold": 0.5
  },
  "logging": {
    "log_dir": "logs/",
    "checkpoint_dir": "/kaggle/working/checkpoints",
    "save_every_n_epochs": 5,
    "log_every_n_steps": 100
  },
  "huggingface": {
    "dataset_repo": "HlexNC/chest-xray-14-320",
    "scratch_model_repo": "HlexNC/chexvision-scratch",
    "transfer_model_repo": "HlexNC/chexvision-densenet",
    "space_repo": "HlexNC/chexvision-demo"
  },
  "model": {
    "type": "densenet",
    "name": "CheXVision-DenseNet",
    "architecture": {
      "backbone": "densenet121",
      "pretrained": true,
      "freeze_features": false,
      "dropout": 0.3
    },
    "fine_tuning": {
      "freeze_epochs": 5,
      "freeze_lr": 0.001,
      "unfreeze_lr": 0.0001
    },
    "heads": {
      "multilabel": {
        "num_classes": 14,
        "activation": "sigmoid"
      },
      "binary": {
        "num_classes": 1,
        "activation": "sigmoid"
      }
    }
  }
}