EKG-mctnet / config.json
SkyAsl's picture
Upload folder using huggingface_hub
b11a412 verified
Raw
History Blame Contribute Delete
314 Bytes
{
"model_type": "MCTnet",
"class_names": [
"AFIB",
"AFL",
"LBBB",
"Normal",
"RBBB"
],
"n_classes": 5,
"seq_len": 5000,
"sampling_rate": 500,
"d_model": 128,
"n_heads": 4,
"n_layers": 4,
"dropout": 0.2,
"ld_emb_dim": 32,
"val_macro_f1": 0.8198,
"framework": "pytorch"
}