{
  "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"
    ]
  }
}