training_config.json

{
  "model_name": "answerdotai/ModernBERT-base",
  "num_labels": 2,
  "label2id": {
    "SAFE": 0,
    "INJECTION_RISK": 1
  },
  "id2label": {
    "0": "SAFE",
    "1": "INJECTION_RISK"
  },
  "batch_size": 32,
  "epochs": 5,
  "learning_rate": 3e-05,
  "max_length": 512,
  "use_class_weights": true,
  "class_weights": [
    0.9990699887275696,
    1.0009300708770752
  ]
}