vulnerability-severity-clf / model_card_data.json
NilHRH's picture
gguf1: canonical ModernBertForSequenceClassification + card
7bd7a10 verified
Raw History Blame Contribute Delete
3.4 kB
{
"run": "gguf1",
"base_model": "jhu-clsp/ettin-encoder-150m",
"classes": [
"Info",
"Low",
"Medium",
"High",
"Critical"
],
"head": "CLS -> Linear(768,768) -> GELU -> LayerNorm -> Dropout -> Linear(768,5)",
"max_len": 320,
"train_config": {
"epochs": 1.0,
"batch": 96,
"lr": 0.0003,
"max_len": 320,
"params_m": 149.6,
"train": 63974,
"val": 3535,
"total_steps": 666,
"head": "ModernBertForSequenceClassification"
},
"test": {
"run": "gguf1",
"split": "test",
"ckpt": "hf_pt",
"n": 3598,
"accuracy": 0.9953,
"macro_f1": 0.9907,
"per_class": {
"Info": {
"support": 163,
"P": 1.0,
"R": 0.9755,
"F1": 0.9876
},
"Low": {
"support": 1072,
"P": 0.9972,
"R": 0.9981,
"F1": 0.9977
},
"Medium": {
"support": 1245,
"P": 0.9944,
"R": 0.9952,
"F1": 0.9948
},
"High": {
"support": 1094,
"P": 0.9936,
"R": 0.9963,
"F1": 0.995
},
"Critical": {
"support": 24,
"P": 1.0,
"R": 0.9583,
"F1": 0.9787
}
},
"confusion": [
[
159,
2,
1,
1,
0
],
[
0,
1070,
2,
0,
0
],
[
0,
1,
1239,
5,
0
],
[
0,
0,
4,
1090,
0
],
[
0,
0,
0,
1,
23
]
]
},
"legacy": {
"run": "enc150",
"test": {
"run": "enc150",
"split": "test",
"ckpt": "best.pt",
"n": 3598,
"accuracy": 0.9956,
"macro_f1": 0.9836,
"per_class": {
"Info": {
"support": 163,
"P": 1.0,
"R": 0.9816,
"F1": 0.9907
},
"Low": {
"support": 1072,
"P": 0.9972,
"R": 1.0,
"F1": 0.9986
},
"Medium": {
"support": 1245,
"P": 0.996,
"R": 0.9952,
"F1": 0.9956
},
"High": {
"support": 1094,
"P": 0.9945,
"R": 0.9945,
"F1": 0.9945
},
"Critical": {
"support": 24,
"P": 0.92,
"R": 0.9583,
"F1": 0.9388
}
},
"confusion": [
[
160,
1,
1,
1,
0
],
[
0,
1072,
0,
0,
0
],
[
0,
2,
1239,
4,
0
],
[
0,
0,
4,
1088,
2
],
[
0,
0,
0,
1,
23
]
]
}
},
"gguf": {
"repo": "NilHRH/vulnerability-severity-clf-GGUF",
"quants": [
"F16",
"Q8_0",
"Q6_K",
"Q5_K_M",
"Q5_K_S",
"Q5_0",
"Q4_K_M",
"Q4_K_S",
"Q4_0",
"Q3_K_L",
"Q3_K_M",
"Q3_K_S",
"Q2_K"
]
}
}