nsrek's picture
Upload folder using huggingface_hub
e4d73f9 verified
Raw
History Blame Contribute Delete
1.23 kB
{
"model_name": "Rocket",
"model_file": "rocket_model.pkl",
"use_case": "OMI vs non-OMI",
"positive_class": "OMI",
"negative_class": "non-OMI",
"class_labels": [
"non-OMI",
"OMI"
],
"features": [
"I",
"aVR",
"V1",
"V4",
"II",
"aVL",
"V2",
"V5",
"III",
"aVF",
"V3",
"V6"
],
"n_features": 12,
"n_timesteps": 141,
"hyperparameters": {
"num_kernels": 50000,
"random_state": 42,
"n_jobs": -1
},
"training_data": "data/ecg_matrix_omi_segmented_50_150_90.csv",
"training_date": "2026-06-02 16:41:15",
"training_time_seconds": 132.78,
"preprocessing": {
"normalization": "per_subject_max_abs",
"smote_tomek": true,
"test_size": 0.25,
"random_state": 42
},
"training_samples": 2084,
"test_samples": 666,
"test_metrics": {
"accuracy": 0.8888888888888888,
"precision": 0.8571428571428571,
"recall": 0.9171974522292994,
"f1": 0.8861538461538462,
"sensitivity": 0.9171974522292994,
"specificity": 0.8636363636363636,
"confusion_matrix": [
[
304,
48
],
[
26,
288
]
]
}
}