config.json

{
  "architectures": [
    "ECGClassifier"
  ],
  "auto_map": {
    "AutoConfig": "modeling_ecg.ECGClassifierConfig",
    "AutoModel": "modeling_ecg.ECGClassifier"
  },
  "model_type": "ecg-classifier",
  "number_of_classes": 5,
  "signal_length": 5000,
  "num_leads": 12,
  "sampling_rate": 100,
  "input_type": "time_series",
  "output_type": "classification",
  "task": [
    "time-series-classification"
  ],
  "classes": [
    "NORM",
    "MI",
    "STTC",
    "HYP",
    "CD"
  ],
  "class_indices": {
    "NORM": 0,
    "MI": 1,
    "STTC": 2,
    "HYP": 3,
    "CD": 4
  },
  "num_layers": 4,
  "output_size": 128,
  "dropout": 0.2,
  "frame_size": 5000,
  "frame_shift": 5000,
  "transformers_version": "4.36.0"
}