poset5-armature-model / config.json
pbk0's picture
Upload config.json with huggingface_hub
5222d9f verified
Raw History Blame Contribute Delete
974 Bytes
{
"model_type": "poset5-cal-compact",
"num_features": 14,
"window_size": 32,
"patch_size": 8,
"d_model": 128,
"num_heads": 4,
"num_layers": 2,
"num_classes": 5,
"key_names": [
"W",
"A",
"S",
"D",
"Space"
],
"best_metrics": {
"val_loss": 0.25981749246517816,
"f1_macro": 0.7377958871424807,
"f1_micro": 0.9314079422382672,
"hamming_loss": 0.03166666666666667,
"per_key": {
"W": {
"f1": 0.9704142011834319,
"prec": 0.9647058823529412,
"rec": 0.9761904761904762
},
"A": {
"f1": 0.8813559322033898,
"prec": 0.8387096774193549,
"rec": 0.9285714285714286
},
"S": {
"f1": 1.0,
"prec": 1.0,
"rec": 1.0
},
"D": {
"f1": 0.8372093023255814,
"prec": 0.9473684210526315,
"rec": 0.75
},
"Space": {
"f1": 0.0,
"prec": 0.0,
"rec": 0.0
}
}
}
}