Upload configs/config.yaml

configs/config.yaml

@@ -0,0 +1,30 @@
+data:
+  dataset_name: "hf-vision/chest-xray-pneumonia"
+  dataset_config: "default"
+  data_dir: "data/cache"
+  image_size: 224
+  batch_size: 32
+  num_workers: 0  # 0 for Windows compatibility
+
+model:
+  name: "mobilenet_v2"  # "simple_cnn" or "mobilenet_v2"
+  num_classes: 2
+  pretrained: true
+  freeze_backbone: true
+  unfreeze_after_epoch: 5
+
+training:
+  epochs: 15
+  learning_rate: 0.001
+  weight_decay: 1.0e-4
+  scheduler: "reduce_on_plateau"
+  patience: 3
+  checkpoint_dir: "checkpoints"
+  seed: 42
+
+evaluation:
+  output_dir: "results"
+
+inference:
+  model_path: "checkpoints/best_mobilenet_v2.pth"
+  threshold: 0.5