Victor Phan
33 lines
563 B
JSON
33 lines
563 B
JSON
{
|
|
"model_type": "Swin-UNet (PyTorch)",
|
|
"input_shape": [
|
|
39
|
|
],
|
|
"embed_dim": 128,
|
|
"num_classes": 8,
|
|
"classes": [
|
|
"Lua tom",
|
|
"Lua",
|
|
"CHN",
|
|
"CLN",
|
|
"TS",
|
|
"Song",
|
|
"Dat xay dung",
|
|
"Rung"
|
|
],
|
|
"label_mapping": {
|
|
"Lua tom": "0",
|
|
"Lua": "1",
|
|
"CHN": "2",
|
|
"CLN": "3",
|
|
"TS": "4",
|
|
"Song": "5",
|
|
"Dat xay dung": "6",
|
|
"Rung": "7"
|
|
},
|
|
"num_parameters": 958536,
|
|
"accuracy": 0.8539823008849557,
|
|
"precision": 0.8893103005023856,
|
|
"recall": 0.8539823008849557,
|
|
"f1_score": 0.8666980102256664
|
|
} |