Modelpublic

rootfs/function-call-sentinel

sourceHugging Faceapache-2.0updated 10mo agoView on Hugging Face

0likes48downloads

training_config.json21 linesDownload Raw Back to root

1{2  "model_name": "answerdotai/ModernBERT-base",3  "num_labels": 2,4  "label2id": {5    "SAFE": 0,6    "INJECTION_RISK": 17  },8  "id2label": {9    "0": "SAFE",10    "1": "INJECTION_RISK"11  },12  "batch_size": 32,13  "epochs": 5,14  "learning_rate": 3e-05,15  "max_length": 512,16  "use_class_weights": true,17  "class_weights": [18    0.9990699887275696,19    1.000930070877075220  ]21}