ECGLight / models /classifier_models /ecg_surgery /model_metadata.json
nsrek's picture
Upload folder using huggingface_hub
e4d73f9 verified
Raw
History Blame Contribute Delete
1.34 kB
{
"model_name": "InceptionTime",
"model_file": "inceptiontime_model.pkl",
"use_case": "Pre vs Post Procedural MI (Surgery)",
"positive_class": "pre-procedural MI",
"negative_class": "post-procedural MI",
"class_labels": [
"post-procedural MI",
"pre-procedural MI"
],
"features": [
"I",
"aVR",
"V1",
"V4",
"II",
"aVL",
"V2",
"V5",
"III",
"aVF",
"V3",
"V6"
],
"n_features": 12,
"n_timesteps": 140,
"hyperparameters": {
"n_epochs": 200,
"batch_size": 64,
"random_state": 42,
"verbose": true
},
"training_data": "data/ecg_surgery_segmented_50_150_70.csv",
"training_date": "2026-06-02 20:26:15",
"training_time_seconds": 251.83,
"preprocessing": {
"normalization": "per_subject_max_abs",
"smote_tomek": true,
"test_size": 0.25,
"random_state": 42
},
"training_samples": 1104,
"test_samples": 338,
"test_metrics": {
"accuracy": 0.9142011834319527,
"precision": 0.8974358974358975,
"recall": 0.9150326797385621,
"f1": 0.9061488673139159,
"sensitivity": 0.9150326797385621,
"specificity": 0.9135135135135135,
"confusion_matrix": [
[
169,
16
],
[
13,
140
]
]
}
}