config.json

{
  "model_type": "simple_cnn",
  "num_classes": 6,
  "labels": [
    "NORM",
    "AFIB",
    "SBRAD",
    "STACH",
    "BIGEMINY",
    "VT"
  ],
  "id2label": {
    "0": "NORM",
    "1": "AFIB",
    "2": "SBRAD",
    "3": "STACH",
    "4": "BIGEMINY",
    "5": "VT"
  },
  "label2id": {
    "NORM": 0,
    "AFIB": 1,
    "SBRAD": 2,
    "STACH": 3,
    "BIGEMINY": 4,
    "VT": 5
  },
  "input_size": [
    3,
    224,
    224
  ],
  "created_at": "2025-09-19T15:23:09.646263"
}